Fix errors

2021-07-02 09:19:39 +00:00
parent 437f62b055
commit 9136f33684
3 changed files with 12 additions and 3 deletions
--- a/exps/basic/xmain.py
+++ b/exps/basic/xmain.py
@@ -74,8 +74,11 @@ def main(args):
    logger.log("The optimizer is:\n{:}".format(optimizer))
    logger.log("The objective is {:}".format(objective))
    logger.log("The metric is {:}".format(metric))
-    logger.log("The iters_per_epoch = {:}, estimated epochs = {:}".format(
-      iters_per_epoch, args.steps // iters_per_epoch))
+    logger.log(
+        "The iters_per_epoch = {:}, estimated epochs = {:}".format(
+            iters_per_epoch, args.steps // iters_per_epoch
+        )
+    )

    model, objective = torch.nn.DataParallel(model).cuda(), objective.cuda()
    scheduler = xmisc.LRMultiplier(