search

Zyun-Y / DconnNet

Codes for CVPR2023 paper "Directional Connectivity-based Segmentation of Medical Images"
138 stars 7 forks source link

Questions about training time. #8

Closed yiseni closed 1 year ago

yiseni commented 1 year ago

I reproduced the model of this article and loaded my own data set for verification. There are 3,062 samples in total. The batch-size is set to 8, and it takes up to an hour to train an epoch. Is it normal? How long does it take for you to train an epoch? Thank you very much.

Sent from PPHub

Zyun-Y commented 1 year ago

May I ask what is your image size? And did you use GPU for training? I don't remember the exact duration of training one epoch. But take Retouch as example, it takes about 5-6 hours in total to train 45 epochs.

I will double check.

yiseni commented 1 year ago

May I ask what is your image size? And did you use GPU for training? I don't remember the exact duration of training one epoch. But take Retouch as example, it takes about 5-6 hours in total to train 45 epochs.

I will double check.

My data size is 640*640, and I use p100 (16g) for training. Thank you