unet++实战学习——Ⅱ train.py debug

指定参数

--dataset dsb2018_96 
--arch NestedUNet

 主要函数

1. 网络结构—— def parse_args():
2. 训练函数—— def train(config ,train_loader, moder, criterion,optimizer)
3. 合法化定义—— def validata(config, val_loader, model ,criterion):
4. 主函数—— def main():

程序运行

由主函数进入网络结构这个大函数下，设置网络。

---

由于这个是在2022年看的，好多东西已经遗忘，后面再用到这个代码在进行补充改进