Skip to content

GitLab

Commit 15e9d026 authored by Kai Chen's avatar Kai Chen
Browse files

remove args from batch_processor

parent 879a6ae4
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tools/train.py
View file @ 15e9d026
...@@ -34,7 +34,7 @@ def parse_losses(losses): ...@@ -34,7 +34,7 @@ def parse_losses(losses):
return loss, log_vars return loss, log_vars
def batch_processor(model, data, train_mode, args=None): def batch_processor(model, data, train_mode):
losses = model(**data) losses = model(**data)
loss, log_vars = parse_losses(losses) loss, log_vars = parse_losses(losses)
...@@ -115,7 +115,7 @@ def main(): ...@@ -115,7 +115,7 @@ def main():
runner.resume(cfg.resume_from) runner.resume(cfg.resume_from)
elif cfg.load_from: elif cfg.load_from:
runner.load_checkpoint(cfg.load_from) runner.load_checkpoint(cfg.load_from)
runner.run(data_loaders, cfg.workflow, cfg.total_epochs, args=args) runner.run(data_loaders, cfg.workflow, cfg.total_epochs)
if __name__ == '__main__': if __name__ == '__main__':
......
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