ICML_2021-源码

WHY FLATNESS与深层网络广义化相关的示例代码 对于我们在带有开放数据集的正文中进行的所有实验，我们提供源代码以重现结果。 这包括数字。 3来图。 主文件中的第5条，附录中的各种结果，对代码的修改最少。 环境： 平台：Linux GPU：NVIDIA GeForce RTX 2080 Ti CPU：我们的匿名计算集群 我们建议您使用自己的python3虚拟环境。 只需创建并激活自己的虚拟环境，然后运行： python3 -m pip install -r requirements.txt 。 然后，您都准备好了！ 复制图。 图3（a），（b），（d），（e）和图。 4，请参阅 复制图。 3（c），（f），请参阅 复制图。 5，请参阅 1和3可以在短时间内在单个CPU上运行，而2和3最好在GPU上运行。