Update README.md

Update to v0.2 checkpoint (#14)

Update code example in readme to use `device` variable instead of hard coded `cuda` (#9)

Update README.md (#6)

Update README.md

Update README.md

Update README.md (#3)

Update README.md

Update README.md

Upload ONNX weights (#1)

Update README.md

End of training

Model save

End of training

Model save

initial commit