ZJF-Thunder
添加文件
e26e560
"""
路径/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)]