rinongal / textual_inversion

MIT License
2.9k stars 279 forks source link

UserWarning: Trying to infer the `batch_size` from an ambiguous collection. The batch size we found is 14. To avoid any miscalculations, use `self.log(..., batch_size=batch_size)`. #87

Closed EKZERSIS closed 2 years ago

EKZERSIS commented 2 years ago
python main.py --base configs/stable-diffusion/v1-finetune.yaml -t --actual_resume models/ldm/text2img-large/modelSD.ckpt -n artcat --gpus 0, --data_root img/datasets/cat_statue --init_word art --placeholder_tokens "*art cat" --no-test

Warn:

UserWarning: Trying to infer the `batch_size` from an ambiguous collection. The batch size we found is 14. To avoid any miscalculations, use `self.log(..., batch_size=batch_size)`.
rinongal commented 2 years ago

This is fine. You can ignore this warning. It should go away after the first epoch.