akanimax / akanimax/big-discriminator-batch-spoofing-gan
Training not progressing 1024x1024 on multigpu
- Linguagem predominante
- Python
- Estrelas
- 45
- Forks
- 7
- Métricas de merge de PRs
- Nenhum PR com merge em 30d
Descrição
I run a few hours on 8GPUs wihtout any progress. Each sample is pixelwise copy of each other in all layers.
4x4

64x64

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
```
Guia de contribuição
Nenhum guia de contribuição indexado para este repositório
Avaliação
Esta issue ainda não foi avaliada.