Zhengxinyang / LAS-Diffusion

MIT License
217 stars 15 forks source link

training time??? #13

Open luckyday2022 opened 10 months ago

luckyday2022 commented 10 months ago

What machine is used for model training? How long is the training time?

Zhengxinyang commented 10 months ago

For each stage, it takes around 3 days for training on 8 V100 Nvidia GPUs.

luckyday2022 commented 10 months ago

The weight file for VIT_MODEL in the prepare_sketch.py file seems to be mismatched with the model.

Zhengxinyang commented 10 months ago

What do you mean by mismatch?

yosun commented 1 month ago

can you put this on replicate or huggingface?