Skip to content

Torch Training Template v2.3

Choose a tag to compare

@LEON-REIN LEON-REIN released this 20 Nov 15:43
· 7 commits to v2 since this release
7469d65
  1. 修复一些 bug: epoch 显示问题, 保存间隔, resume 逻辑 ......
  2. 完善 Python Type Hint 支持
  3. 支持默认读取组件的行为, 尤其可以自适应加载零个或多个测试集进行测试
  4. 代码优化, 用 @torch.inference_mode() 代替 with torch.no_grad, 防止多余的缩进, 等