Spaces:
Running
Running
File size: 676 Bytes
499e141 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
# -*- coding: utf-8 -*-
# @Author : xuelun
from yacs.config import CfgNode as CN
_CN = CN()
# ------------
# Trainer
# ------------
_CN.TRAINER = CN()
_CN.TRAINER.NUM_SANITY_VAL_STEPS = 0
_CN.TRAINER.LOG_INTERVAL = 1
_CN.TRAINER.VAL_CHECK_INTERVAL = 1.0 # default 1.0, if we set 2.0 will val each 2 step
_CN.TRAINER.LIMIT_TRAIN_BATCHES = 10.0 # default 1.0
_CN.TRAINER.LIMIT_VALID_BATCHES = 10.0 # default 1.0 will use all training batch
def get_cfg_defaults():
"""Get a yacs CfgNode object with default values for my_project."""
# Return a clone so that the defaults will not be altered
# This is for the "local variable" use pattern
return _CN.clone()
|