- Clone this repository
git clone https://github.com/magnusbarata/brain-tumor-classification.git cd brain-tumor-classification
- Install the dependencies
docker build -t tf-gpu setup docker run --gpus all -u $(id -u):$(id -g) -it --rm -v $(pwd):/work tf-gpu bash
- Download the data and put it inside
data
directory. - Set the hyperparameter file and start training.
python train.py <EXP_DIR>
- Monitor using tensorboard.
tensorboard --logdir <EXP_DIR>/logs