conda activate bwgan
conda install cudatoolkit
pip install numpy scipy matplotlib tqdm tensorboardx torch torchvision
export LD_LIBRARY_PATH=/home/student/.conda/envs/bwgan/lib
git clone https://github.com/new-okaerinasai/bwgan_pytorch
cd bwgan_pytorch
git checkout old_branch
python train.py --dataset celeba --name s0p2.5 --s 0 --p 2.5 --datapath /home/student/celeba --logspath /home/student/logs_bwgan --cuda 0