segmentation-guided-diffusion icon indicating copy to clipboard operation
segmentation-guided-diffusion copied to clipboard

About training time

Open ashevimostar opened this issue 6 months ago • 1 comments

I use a 3090 card to train for 400 epochs, which takes about a week. May I ask what your device is and how long the training time is? Is there a problem with my training time, is there a way to accelerate, or is there a way to change my parameters.

CUDA_VISIBLE_DEVICES={DEVICES} python3 main.py
--mode train
--model_type DDIM
--img_size {IMAGE_SIZE}
--num_img_channels {NUM_IMAGE_CHANNELS}
--dataset {DATASET_NAME}
--img_dir {DATA_FOLDER}
--train_batch_size 16
--eval_batch_size 8
--num_epochs 400

ashevimostar avatar Jun 04 '25 02:06 ashevimostar