设置/修改nnUNet训练的Epoch、显卡号

一、设置/修改Epoch

由于nnUNet默认的epoch=1000，训练是在太久了，可用以下方式修改训练的epoch的默认值

1、在nnUNet的源码目录中找到nnUNetTrainerV2.py，该文件的位置如下：

2、直接将原来的self.max_num_epochs=1000 改为 自己想要的值，如改为23

二、修改训练时的显卡号

在训练语句前添加CUDA_VISIBLE_DEVICES=N，  N是指定的显卡号。
在不指定的情况下，N默认为0，以下是修改为1号显卡的实例：

CUDA_VISIBLE_DEVICES=1 nnUNet_train 3d_fullres nnUNetTrainerV2 676 2

顺便附上本人在Ubuntu下跑通自己数据集的具体实施过程，链接如下：https://blog.csdn.net/weixin_42649699/article/details/122439094

部分细节没写到位，有疑问可以私信，尽量回复。

码字不易，有参考意义就给个赞呗，谢谢~~