/simpletorchdemo

for torch beginer

Primary LanguagePython

simple torch demo

you can use this repo as a simple codebase

TODO

  • training and evaluation
  • Parse agrgs
  • optimizer
  • model and dataset
  • DDP
  • tensorboard
  • save and resume