11 Commits (72e4b6b0f1286e148454f89bda81ac9cace8d32e)

Author SHA1 Message Date
  shenyan 72e4b6b0f1 设置workers最大数量为8 4 years ago
  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