search

guoqincode / Train_SD_VAE

Finetune your VAE on private datasets!
19 stars 2 forks source link

About mixed precision training #3

Closed qiuzidian closed 3 months ago

qiuzidian commented 3 months ago

Thank you for your script! However, is it currently not possible to perform mixed precision training through the mixed_precision option? I made some changes to the code before it could run properly.

guoqincode commented 3 months ago

Hi, you can implement mixed precision training by configuring accelerator config