Fixed naming bug

main.py

@@ -74,7 +74,7 @@ if __name__ == "__main__":
 
     optimizer, model = get_optimizer(args, model)
 
-    num_steps = ceil(args.epochs * len(dataset) / args.batch_size)
+    num_steps = ceil(args.num_epochs * len(dataset) / args.batch_size)
     scheduler = get_scheduler(
         optimizer, args.lr_scheduler, args.warm_up_fraction, num_steps, args.lr, args.lr_min
     )