machineLearningTemplate

54 kB

Author	SHA1	Message	Date
shenyan	b2ac2ad2c1	增加训练阶段的数据增强; 将输入数据标准化; 删除main中的config不需要的量并修改输入维度;	4 years ago
shenyan	ef272dfb51	删除恒源云的使用代码;修改北京云的训练脚本; main中修改了部分参数并在设置gpu的情况下自动调用gpu; 注明train model中训练步的返回值	4 years ago
shenyan	24fdd1244e	修改main函数的格式	4 years ago
shenyan	370e96c7f7	更换网络为ResNet56; 保存当前版本到最后保存位置	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