dreambooth error

[nokiben]

1\File "train_dreambooth.py", line 11, in import pynvml ModuleNotFoundError: No module named 'pynvml'

2\ValueError: num_samples should be a positive integer value, but got num_samples=0

come from 2 different colab link when start training

[ClashSAN]

@d8ahazard's dreambooth extension was designed for local use, why not just use the other existing standalone colabs for dreambooth? The extension is adapted from Shrivam's implementation. Are you trying to compare results?

[kuratatomoaki]

hi.

in another case
Generate a preview image every N steps, 0 to disable

If you set this to 0

save_img = not global_step % args.save_preview_every and global_step != 0
ZeroDivisionError: integer division or modulo by zero

I get the above error