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上运行。
2021-10-04 21:49:12
65.11MB
Python
1