fix range epochs

0a02d1f2 · ouyangyu · 302fa6a8 · 0a02d1f2
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

Classification/cnns/of_cnn_train_val.py Classification/cnns/of_cnn_train_val.py +1 -1

未找到文件。
--- a/Classification/cnns/of_cnn_train_val.py
+++ b/Classification/cnns/of_cnn_train_val.py
@@ -137,7 +137,7 @@ def main():

    print(" {} iter per epoch...".format(epoch_size))

-    for epoch in range(1, args.num_epochs):
+    for epoch in range(1, args.num_epochs + 1):
        metric = Metric(
            desc="train",
            calculate_batches=args.loss_print_every_n_iter,