StyleDrop-PyTorch

StyleDrop-PyTorch copied to clipboard

yatoubusha

accelerate launch --num_processes 8 --mixed_precision fp16 train_t2i_custom_v2.py --config=configs/custom.py I use the above command to train a picture style model, loss is always shock, grad_norm: nan (nan), the effect is not...

EnigmaHong

image Is this module included in the code? I can't seem to find it.

Laidawang

Error:` list index out of range at line last_subdir = subdirs[-1]` under Training cell. Are there folders that I should manually add?

fypbiqi

trouble running due to memory

1

I have a Nvidia 2060 Super 8GB ... is this enough? I see that the model download is 10GB and I am getting torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to...

ownimage

Poor results with training 1000 steps

1

Training Image: Banana on a table: 

AmanKishore

First of all, thank you for this implementation. I'm hoping you can help. I have a specific set of illustrations and I'd like to generate more in that style. I've...

carldaws

Thanks for your great contribution to this field. May I inquire the training time on A100 GPUs of this project? Thanks~

maxin-cn