18 Commits (master)

 

All Branches   

Search

Author	SHA1	Message	Date
shenyan	67ec993c27	保存版本训练结果时输出备注的版本信息, 主要记录该版本的网络数据集等.	4 years ago
shenyan	72e4b6b0f1	设置workers最大数量为8	4 years ago
shenyan	b2ac2ad2c1	增加训练阶段的数据增强; 将输入数据标准化; 删除main中的config不需要的量并修改输入维度;	4 years ago
shenyan	1a12a8c27b	使用warm up的学习率调整策略	4 years ago
shenyan	ef272dfb51	删除恒源云的使用代码;修改北京云的训练脚本; main中修改了部分参数并在设置gpu的情况下自动调用gpu; 注明train model中训练步的返回值	4 years ago
shenyan	665bca9171	考虑保存版本信息时分割字符为'/'的情况; 删除读取表格相关的内容;	4 years ago
shenyan	76b29f1a8f	更改requirements版本; 改变保存版本信息时的最优值的获取逻辑; 修复保存版本信息时可能产生的BUG	4 years ago
shenyan	24fdd1244e	修改main函数的格式	4 years ago
shenyan	370e96c7f7	更换网络为ResNet56; 保存当前版本到最后保存位置	4 years ago
shenyan	d68c76b4fd	修改visual_label中创建多级目录的BUG	4 years ago
shenyan	f23c6ce854	修改得到数据list的方式	4 years ago
shenyan	6e1eecfc30	改变产生拟合函数数据的方式;	4 years ago
shenyan	436625c6ff	使用较为复杂的拟合网络; 将训练结果输出至文件;	4 years ago
shenyan	ea7bd5d762	修改根据版本号和当前fold数得到ckpt地址的函数的BUG; 加入基于JDLU激活函数的网络; 改变网络的初始化方	4 years ago
shenyan	d7bfcf0830	修改了evaluate中的拼写错误;使用自定义激活函数jdlu; 添加由版本名获得对应版本的ckpt的函数;改变存在保存路径的情况下的保存模式; 开发k折交叉验证; 增加了excel表格读写函数	4 years ago
shenyan	b114979ea5	改用pin_memory=True;添加对gpu性能的注意说明;自动获取核心数并设置其为num_workers;简化每个epoch的随机数更换逻辑	4 years ago
shenyan	4222cd2770	将模型改为拟合函数; 可视化拟合函数的结果	4 years ago
shenyan	52ed4770bb	完成神经网络训练框架; 完成分类任务	4 years ago