Python中导入模块中的函数运行出错

Question

Python中将deeplabv3+训练过程当做一个模块被调用时出错

deeplabv3+训练模块单独运行时不报错，被另一个文件调用时报错

train. py的内容:

import os
import datetime
import torch
省略掉一些导入的代码
def TrainDeeplabV3Plus():
  省略
if __name__=="__main__":
    TrainDeeplabV3Plus()

test. py的内容：

import train
train.TrainDeeplabV3Plus()

执行train.py是没问题的可以正常运行：

执行test. py报错而且很怪:

像是没跑完又重新从头开始加载了一样

应该不是import的问题，毕竟test.py中的内容也会运行，只不过是运行结果不对

等智慧的人解答，要更详细的内容可以call我，悬赏！