Open fikry102 opened 1 year ago
The paper says "The training process takes about 20 minutes each image on one NVIDIA GeForce RTX3090 with a batch size of 1." I used two RTX3090. However, it already cost three hours and it doesn't seem to stop.
For "v1-finetune.yaml”, which paramater specifys the number of epochs in the training process?
As for me, if you dont not Ctrl+C, with the default yaml, it won't be stop
In main.py, change trainer_kwargs["max_steps"] = opt.max_steps to trainer_kwargs["max_steps"] = trainer_opt.max_steps you will can use max_steps in the v1-finetune.yaml file
The paper says "The training process takes about 20 minutes each image on one NVIDIA GeForce RTX3090 with a batch size of 1." I used two RTX3090. However, it already cost three hours and it doesn't seem to stop.
For "v1-finetune.yaml”, which paramater specifys the number of epochs in the training process?