深度学习检查当前处于测试模式还是训练模式

import argparse

# 解析命令行参数
parser = argparse.ArgumentParser()
parser.add_argument('--isTrain', action='store_true', help='Set to True for training mode')
opt = parser.parse_args()

# 检查是否为训练模式
if opt.isTrain:
    print("当前处于训练模式")
else:
    print("当前处于测试模式")