""" | |
路径/configs/base/default_runtime.py中: | |
第1行interval=1表示每1个epoch保存一次权重信息 | |
第4行interval=50表示每50次打印一次日志信息 | |
第14行load_from表示加载训练好的权重路径,可以不设置,在训练时终端中给定 | |
""" | |
# 表示多少个 epoch 验证一次,然后保存一次权重信息 | |
# interval=1 表示每1个epoch保存一次权重信息 | |
checkpoint_config = dict(interval=25) | |
# yapf:disable | |
log_config = dict( | |
interval=20, # 表示20个 epoch 打印一次日志信息 | |
hooks=[ | |
dict(type='TextLoggerHook'), | |
# dict(type='TensorboardLoggerHook') | |
]) | |
# yapf:enable | |
custom_hooks = [dict(type='NumClassCheckHook')] | |
dist_params = dict(backend='nccl') | |
log_level = 'INFO' | |
# loadfrom:表示加载哪一个训练好的权重 | |
load_from = None | |
# load_from = r"checkpoints/latest.pth" | |
# load_from = r"checkpoints/mask_rcnn_swin_tiny_patch4_window7.pth" | |
# load_from = r"D:\Coding\University Course Study\CV project\Swin-Transformer-Object-Detection\mask_rcnn_swin_tiny_patch4_window7.pth" | |
resume_from = None | |
workflow = [('train', 1)] | |