Commit 35df959b authored by Shaoshuai Shi's avatar Shaoshuai Shi
Browse files

set to only evaluate last epoch in train.py

parent abd7b7f0
......@@ -186,7 +186,7 @@ def main():
)
eval_output_dir = output_dir / 'eval' / 'eval_with_train'
eval_output_dir.mkdir(parents=True, exist_ok=True)
args.start_epoch = max(args.epochs - 10, 0) # Only evaluate the last 10 epochs
args.start_epoch = max(args.epochs - 0, 0) # Only evaluate the last 10 epochs
repeat_eval_ckpt(
model.module if dist_train else model,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment