67ec993
(HEAD -> master)
保存版本训练结果时输出备注的版本信息, 主要记录该版本的网络数据集等. by
2021-10-22 11:46:55 +0800
72e4b6b
设置workers最大数量为8 by
2021-10-22 10:26:43 +0800
b2ac2ad
增加训练阶段的数据增强; 将输入数据标准化; 删除main中的config不需要的量并修改输入维度; by
2021-10-21 19:55:25 +0800
1a12a8c
使用warm up的学习率调整策略 by
2021-10-21 16:46:45 +0800
ef272df
删除恒源云的使用代码;修改北京云的训练脚本; main中修改了部分参数并在设置gpu的情况下自动调用gpu; 注明train model中训练步的返回值 by
2021-10-21 12:36:52 +0800
665bca9
考虑保存版本信息时分割字符为'/'的情况; 删除读取表格相关的内容; by
2021-10-16 17:03:10 +0800
76b29f1
更改requirements版本; 改变保存版本信息时的最优值的获取逻辑; 修复保存版本信息时可能产生的BUG by
2021-10-16 16:46:39 +0800
24fdd12
修改main函数的格式 by
2021-10-15 20:29:55 +0800
370e96c
更换网络为ResNet56; 保存当前版本到最后保存位置 by
2021-10-15 20:28:11 +0800
d68c76b
修改visual_label中创建多级目录的BUG by
2021-10-15 19:26:18 +0800
f23c6ce
修改得到数据list的方式 by
2021-10-15 19:24:56 +0800
6e1eecf
改变产生拟合函数数据的方式; by
2021-10-15 16:33:23 +0800
436625c
使用较为复杂的拟合网络; 将训练结果输出至文件; by
2021-10-15 11:16:20 +0800
ea7bd5d
修改根据版本号和当前fold数得到ckpt地址的函数的BUG; 加入基于JDLU激活函数的网络; 改变网络的初始化方 by
2021-10-14 15:36:01 +0800
d7bfcf0
修改了evaluate中的拼写错误;使用自定义激活函数jdlu; 添加由版本名获得对应版本的ckpt的函数;改变存在保存路径的情况下的保存模式; 开发k折交叉验证; 增加了excel表格读写函数 by
2021-10-14 15:33:59 +0800
b114979
改用pin_memory=True;添加对gpu性能的注意说明;自动获取核心数并设置其为num_workers;简化每个epoch的随机数更换逻辑 by
2021-10-14 15:32:17 +0800
4222cd2
将模型改为拟合函数; 可视化拟合函数的结果 by
2021-10-14 15:31:40 +0800
52ed477
完成神经网络训练框架; 完成分类任务 by
2021-10-14 15:29:39 +0800