search

lee-zq / 3DUNet-Pytorch

3DUNet implemented with pytorch
484 stars 108 forks source link

关于模型训练的过程中遇到的问题 #38

Open MD7sjh opened 9 months ago

MD7sjh commented 9 months ago

尊敬的作者您好，我想问一下，当我复现您的代码一直到训练的时候，我发现我的终端虽然会显示训练的进度条和epoch，但是过了一段时间后仅仅是显示但是并没有任何进展，这个是因为该模型训练一次epoch需要很长时间呢还是哪里出了问题？期待您的回复

LambdaLi commented 9 months ago

我用3080ti，一个epoch才1.30min

MD7sjh commented 9 months ago

我用3080ti，一个epoch才1.30min

那您的batch_size是多少来训练的呢