From 15e9d0263386076f4c37024921d588871257d288 Mon Sep 17 00:00:00 2001
From: Kai Chen <chenkaidev@gmail.com>
Date: Sun, 23 Sep 2018 23:05:59 +0800
Subject: [PATCH] remove args from batch_processor
---
tools/train.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/tools/train.py b/tools/train.py
index 742bd1b..2111546 100644
--- a/tools/train.py
+++ b/tools/train.py
@@ -34,7 +34,7 @@ def parse_losses(losses):
return loss, log_vars
-def batch_processor(model, data, train_mode, args=None):
+def batch_processor(model, data, train_mode):
losses = model(**data)
loss, log_vars = parse_losses(losses)
@@ -115,7 +115,7 @@ def main():
runner.resume(cfg.resume_from)
elif 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__':
--
GitLab