文件名称:iclr-2021-baynne:ICLR 2021深度学习仿真研讨会论文
文件大小:27.79MB
文件格式:ZIP
更新时间:2024-04-26 17:10:46
Python
iclr-2021-Baynne ICLR 2021仿真深度学习研讨会论文。 档案文件 main.py:主文件,设置超参数,执行训练,保存模型检查点等 BNN.py:异方差神经网络的类定义 gaussian_nll_loss_class.py:高斯负对数似然损失 anchor_loss.py:锚点丢失 RMSE.py:均方根误差损失 基本用法 训练单个神经网络 python main.py(默认值:种子= 0,开始时期= 0) python main.py -s 42(使用种子= 42播种训练过程) python main.py -s 42 -c 1000(从预先保存为epoch = 1000的模型开始训练过程) 合奏训练 可以通过唯一地播种该合奏的每个成员并运行'python main.py -s(seed)'来训练一个合奏。 要求 python 3.6.8 pytorch 1.
【文件预览】:
iclr-2021-baynne-main
----.gitignore(9B)
----gaussian_nll_loss_class.py(1KB)
----RMSE.py(257B)
----LICENSE(1KB)
----README.md(818B)
----anchor_loss.py(771B)
----output_1()
--------mymodel_1_1000(31.02MB)
----main.py(5KB)
----BNN.py(2KB)