pytorch-animeGAN
pytorch-animeGAN copied to clipboard
Where is the weight of the discriminator, please?
==== Train Config ====
dataset Hayao data_dir /content/dataset epochs 100 init_epochs 4 batch_size 6 checkpoint_dir {ckp_dir} save_image_dir {save_img_dir} gan_loss lsgan resume GD use_sn True save_interval 1 debug_samples 0 lr_g 2e-05 lr_d 4e-05 init_lr 0.0001 wadvg 10.0 wadvd 10.0 wcon 1.5 wgra 3.0 wcol 30.0 d_layers 3 d_noise False
Init models...
Compute mean (R, G, B) from 1792 images
100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 1792/1792 [00:04<00:00, 373.49it/s]
Mean(B, G, R) of Hayao are [-4.4346958 -8.66591597 13.10061177]
Dataset: real 6656 style 1792, smooth 1792
G weight loaded
Could not load checkpoint, train from scratch [Errno 2] No such file or directory: '{ckp_dir}/discriminator_Hayao.pth'
Epoch 13/100
0%| | 0/1110 [00:06<?, ?it/s]
Traceback (most recent call last):
File "train.py", line 247, in
the error is not about discriminator weight, please check again your path, the actually error is
RuntimeError: CUDA error: no kernel image is available for execution on the device
I understand, thank you very much!
FileNotFoundError: Dataset not found /content/dataset/Kimetsu
Is there a problem with the file path in this case?