train speed seems too slow
Hazarch opened this issue · 1 comments
Hazarch commented
On A800 80G, seting train_batch_size = 20, it seems consume 6 mins using fp16.
lawrence-cj commented
I would suggest you to pre-compute all the T5 and VAE features in advance for fast training.