After miserable experience on running training on this repo. I think I need to write it down to share to others. My situation is: I failed run the model on my own ubuntu server, as I didn't install CUDA10.0. But, running in the docker is also not so easy.
Run the demo
Build the docker
docker build -t new-comod-gan .
docker run -itd -v /your_work_dir:/work -v /your_data_dir:/data --name comod -p 7200-7220:7200-7220 --gpus all new-comod-gan /bin/bash
After miserable experience on running training on this repo. I think I need to write it down to share to others. My situation is: I failed run the model on my own ubuntu server, as I didn't install CUDA10.0. But, running in the docker is also not so easy.
Run the demo
.bashrc
Remote X11
plugin in vscode.Host
Option, and set your remote Host'IP with your remote server's IP(Not the container's IP)Run training
Preparing your own training dataset Here, I prepared some images in
./imgs/png_samples/
for training test.Note:
--num-channels
to 4, you'll get error in training.--val-image-dir
should be specified, or you'll have error in training.Run your training
For researchers in China, you may need a VPN. The training process will download an inception model file. You can: