Run in Colab

[MyaMashudi]

Hi. Can light-weight GAN runs in Colab?

[woctezuma]

Yes.

[MyaMashudi]

Yes.

I have tried running this code: ! lightweight_gan --data ./path/to/images --image-size 512

But it took more than a day and it is stuck between 60% to 70%. I have tried several times running the code, the problem is still the same.

[Dok11]

After 0.16.1 I sometimes catch stucks (with 100% usage of GPU) but rerun helps to me. On my PC, not Colab.

[MyaMashudi]

After 0.16.1 I sometimes catch stucks (with 100% usage of GPU) but rerun helps to me. On my PC, not Colab.

I'm still running it, but I reduce the batch-size. Hope it is going to work fine.

[Dok11]

I'm still running it, but I reduce the batch-size. Hope it is going to work fine.

As I think the batch size dont relative to stucks. Reducing the batch size is valid only if GPU memory dont enough, keep batch size as large as you can