Deeplearning 구현 연습을 위한 repo입니다.
Install dependencies
# clone project
git clone https://github.com/hrnoh24/deeplearning-examples.git
cd deeplearning-examples
# build docker
scripts/docker_build.sh
# run docker interactive
scripts/docker_interactive.sh [dataset path] # dataset path 지정 시 "/data"로 mount
Train model with default configuration
# train on CPU
python src/train.py trainer=cpu
# train on GPU
python src/train.py trainer=gpu
Train model with chosen experiment configuration from configs/experiment/
python src/train.py experiment=experiment_name.yaml
You can override any parameter from command line like this
python src/train.py trainer.max_epochs=20 datamodule.batch_size=64