Fix benchmark to use standard epochs argument

Author: jmohdyusof

Pilot3/P3B6/default_model.txt

@@ -4,7 +4,7 @@ learning_rate = 2e-5
 eps = 1e-8
 weight_decay = 0.0
 batch_size = 10
-num_epochs = 10
+epochs = 10
 rng_seed = 13
 num_train_samples = 10000
 num_valid_samples = 10000

Pilot3/P3B6/p3b6.py

@@ -30,7 +30,7 @@
     'weight_decay',
     'rng_seed',
     'batch_size',
-    'num_epochs',
+    'epochs',
 ]
 
 

Pilot3/P3B6/p3b6_baseline_pytorch.py

@@ -134,7 +134,7 @@ def run(args):
 
     criterion = nn.BCEWithLogitsLoss()
 
-    for epoch in range(args.num_epochs):
+    for epoch in range(args.epochs):
         train(train_loader, model, optimizer, criterion, args, epoch)
         validate(valid_loader, model, args, epoch)