/Hole5DoF

For BMI task

Primary LanguagePython

BMI task to find 5DoF holes

Installation

Python>=3.7, Install libraries by pip install -r requirements.txt.

Train MNIST

Run (example)

python train.py --root_dir "./"

You can monitor the training process by launching tensorboard in another terminal:

tensorboard --logdir logs

And go to localhost:6006 in your browser.