Graphic Project Training command python train.py --dataset dataset Test command python test.py --model path/to/module