You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
shenyan 67ec993c27 保存版本训练结果时输出备注的版本信息, 主要记录该版本的网络数据集等. 4 years ago
free_servers 删除恒源云的使用代码;修改北京云的训练脚本; main中修改了部分参数并在设置gpu的情况下自动调用gpu; 注明train model中训练步的返回值 4 years ago
network 更换网络为ResNet56; 保存当前版本到最后保存位置 4 years ago
network_module 完成神经网络训练框架; 完成分类任务 4 years ago
custom_profile.py 完成神经网络训练框架; 完成分类任务 4 years ago
data_module.py 增加训练阶段的数据增强; 将输入数据标准化; 删除main中的config不需要的量并修改输入维度; 4 years ago
evaluate.py 修改了evaluate中的拼写错误;使用自定义激活函数jdlu; 添加由版本名获得对应版本的ckpt的函数;改变存在保存路径的情况下的保存模式; 开发k折交叉验证; 增加了excel表格读写函数 4 years ago
main.py 保存版本训练结果时输出备注的版本信息, 主要记录该版本的网络数据集等. 4 years ago
requirements.txt 更改requirements版本; 改变保存版本信息时的最优值的获取逻辑; 修复保存版本信息时可能产生的BUG 4 years ago
save_checkpoint.py 保存版本训练结果时输出备注的版本信息, 主要记录该版本的网络数据集等. 4 years ago
train_model.py 使用warm up的学习率调整策略 4 years ago
utils.py 考虑保存版本信息时分割字符为'/'的情况; 删除读取表格相关的内容; 4 years ago

基于pytorch lightning的机器学习模板, 用于对机器学习算法进行训练, 验证, 测试等, 目前实现了神经网路, 深度学习, k折交叉, 自动保存训练信息等.

Python Text Shell

Contributors (1)