machineLearningTemplate

54 kB

Author	SHA1	Message	Date
shenyan	b2ac2ad2c1	增加训练阶段的数据增强; 将输入数据标准化; 删除main中的config不需要的量并修改输入维度;	4 years ago
shenyan	370e96c7f7	更换网络为ResNet56; 保存当前版本到最后保存位置	4 years ago
shenyan	f23c6ce854	修改得到数据list的方式	4 years ago
shenyan	6e1eecfc30	改变产生拟合函数数据的方式;	4 years ago
shenyan	436625c6ff	使用较为复杂的拟合网络; 将训练结果输出至文件;	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