Open luckyday2022 opened 10 months ago
For each stage, it takes around 3 days for training on 8 V100 Nvidia GPUs.
The weight file for VIT_MODEL in the prepare_sketch.py file seems to be mismatched with the model.
What do you mean by mismatch?
can you put this on replicate or huggingface?
What machine is used for model training? How long is the training time?