akanimax / akanimax/big-discriminator-batch-spoofing-gan

Training not progressing 1024x1024 on multigpu

Open
#2 5 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
45
Forks
7
PR merge metrics
No merged PRs in 30d

Description

I run a few hours on 8GPUs wihtout any progress. Each sample is pixelwise copy of each other in all layers.

4x4
![image](https://user-images.githubusercontent.com/37534415/66875587-88854600-efb7-11e9-8058-d8bca9c0697a.png)

64x64
![image](https://user-images.githubusercontent.com/37534415/66875607-9a66e900-efb7-11e9-9ce2-5a48b4c02299.png)

setup
```
git clone git@github.com:akanimax/BBMSG-GAN.git
conda create -n bbmsg python==3.7
conda activate bbmsg
conda install pytorch torchvision cudatoolkit=10.0 cudnn scipy==1.2.0 tensorboard -c pytorch
pip install tensorboardX tqdm
```

training
```
# calc real fid stats before training
python ../BBMSG-GAN/sourcecode/train.py \
--images_dir="$IMGS" \
--sample_dir="$SAMPLES" \
--model_dir="$MODELS" \
--depth=9 \
--batch_size=24 \
--num_samples=36 \
--feedback_factor=5 \
--checkpoint_factor=1 \
--num_epochs=50000 \
--num_workers=90 \
--log_fid_values=True \
--fid_temp_folder=/tmp/fid_tmp \
--fid_real_stats="$FID" \
--fid_batch_size=64 \
--num_fid_images=5000
```

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.