コード例 #1
0
 def evaluate_epoch(engine):
     log_results(logger, 'train/epoch', engine.state, engine.state.epoch)
     state = evaluate_once(evaluator, iterator=iters['val'])
     log_results(logger, 'valid/epoch', state, engine.state.epoch)
     log_results_cmd('valid/epoch', state, engine.state.epoch)
     save_ckpt(args, engine.state.epoch, engine.state.metrics['loss'], model, vocab)
     evaluate_by_logic_level(args, model, iterator=iters['val'])
コード例 #2
0
 def evaluate_epoch(engine):
     log_results(logger, 'train/epoch', engine.state, engine.state.epoch)
     state = evaluate_once(evaluator, iterator=iters['val'])
     log_results(logger, 'valid/epoch', state, engine.state.epoch)
コード例 #3
0
 def log_iter_results(engine):
     log_results(logger, 'train/iter', engine.state, engine.state.iteration)
コード例 #4
0
 def evaluate_epoch(engine):
     log_results(logger, 'pretrain/epoch', engine.state, engine.state.epoch)
コード例 #5
0
ファイル: pretrain.py プロジェクト: sally20921/ConSSL
 def save_epoch(engine):
     log_results(logger, 'pretrain/epoch', engine.state, engine.state.epoch)
     log_results_cmd(logger, 'pretrain/epoch', engine.state, engine.state.epoch)
     save_ckpt(args, engine.state.epoch, engine.state.metrics['loss'], model)