nvidia-docker run --name GAN -it -v /your/project/path/:/project/path/ --shm-size=64g nvcr.io/nvidia/pytorch:20.11-py3
apt update
apt install -y libgl1-mesa-glx
pip uninstall opencv-python
pip install opencv-python
you can change parameters using config.py
cd /project/path
python train.py
GAN에 비해서 어느정도 완화되긴 했지만 생성기와 분별기의 학습 비율 조정에 어려움이 존재한다.