nnUNetV2修改笔记-D6

开始大规模训练

现在已经完成的

  • 原始nnUNet

  • 原始nnUNet + EAB模块

  • 原始STUNet Base

  • 原始STUNet Small

正在训练的

  • 原始nnUNet + ChannelAttention

Z8的RTX3060 6G Laptop正在训练,大概530s一个Epoch。(06-22 12:00)
温度有点高。
PS:买的2080Ti 22G到了,周一装上去,应该就爽了。
3060.PNG

  • 原始STUNet Base + TotalSegmentator预训练模型

公司RTX3090Ti正在训练,目前训练到(F3,50E)大概300s一个Epoch。(06-22 12:00)
3090.PNG

1
nnUNetv2_train 43 3d_fullres -tr STUNetTrainer_base -pretrained_weights "D:\wzh\base_ep4k_v2.pth" 0 --npz; nnUNetv2_train 43 3d_fullres -tr STUNetTrainer_base 4 --npz; nnUNetv2_train 43 3d_fullres -tr STUNetTrainer_base -pretrained_weights "D:\wzh\base_ep4k_v2.pth" 1 --npz; nnUNetv2_train 43 3d_fullres -tr STUNetTrainer_base -pretrained_weights "D:\wzh\base_ep4k_v2.pth" 2 --npz; nnUNetv2_train 43 3d_fullres -tr STUNetTrainer_base -pretrained_weights "D:\wzh\base_ep4k_v2.pth" 3 --npz; nnUNetv2_train 43 3d_fullres -tr STUNetTrainer_base -pretrained_weights "D:\wzh\base_ep4k_v2.pth" 4 --npz

下一步要做的事

  • 公司3090Ti慢得离谱,想办法看下是咋回事,速度还不如2080Ti。不知道用WSL2是否会有改善。

  • 验证验证集还没搞好,传上去几天才有回信,我输入的团体名被占用了,还得重新申请,我真是服了。