This is a practice program for the Spring Seminar 2019 in the Robotics Lab of NAIST.
- macOS Mojave 10.14.3
- Python 3.6.8 (Anaconda 4.6.4)
- PyTorch 1.0.0
- TensorboardX 1.6
Run python train.py
.
After running train.py
, a model.pth
file should have been created.
Run test.py
to test the model you have trained.