Open zibingo opened 5 months ago
Hi, to train the model just execute this command:
python main.py --base configs/latent-diffusion/diffusion-sem-with_uc-mask-attn.yaml -t --gpus 0, -n <name of the output>
If for any reason you want to resume a training just use
python main.py --base configs/latent-diffusion/diffusion-sem-with_uc-mask-attn.yaml -t --gpus 0, -r <ckpt folder>
To train the model with custom dataset or parameters you have to change the yaml file and customize the dataset.py file
Hi, to train the model just execute this command:
If for any reason you want to resume a training just use
To train the model with custom dataset or parameters you have to change the yaml file and customize the dataset.py file