Traceback (most recent call last): File "D:\PFL-Non-IID-master\dataset\generate_mnist.py", line 78, in <module> balance = True if sys.argv[2] == "balance" else False IndexError: list index out of range如何解决

这个错误提示也表明你的代码调用了一个空列表里的元素，导致了索引超出范围的错误。同样的，有可能是你在运行代码时没有传递命令行参数，或者传递的参数数量不正确。

你可以检查一下你的命令行参数是否正确传递，如果没有传递，可以在代码中设置默认参数。例如，你可以将代码的第一行修改为：

import sys

niid = True if len(sys.argv) > 1 and sys.argv[1] == "noniid" else False
balance = True if len(sys.argv) > 2 and sys.argv[2] == "balance" else False

这样代码就会首先检查命令行参数是否传递了 "noniid" 和 "balance"，如果没有传递则使用默认值 False。