minor

Committed-by: LiSu from Dev container
mlcommons · Mar 21, 2024 · b98bb57 · b98bb57
1 parent 8ed4fc8
commit b98bb57
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/graph_neural_network/dist_train_rgnn.py b/graph_neural_network/dist_train_rgnn.py
@@ -418,7 +418,7 @@ def run_training_proc(local_proc_rank, num_nodes, node_rank, num_training_procs,
 
     mllogger.event(key=mllog_constants.GLOBAL_BATCH_SIZE, value=world_size*args.train_batch_size)
     mllogger.event(key=mllog_constants.GRADIENT_ACCUMULATION_STEPS, value=1)
-    mllogger.event(key=mllog_constants.OPT_NAME, value='Adam')
+    mllogger.event(key=mllog_constants.OPT_NAME, value='adam')
     mllogger.event(key=mllog_constants.OPT_BASE_LR, value=args.learning_rate)
     mllogger.event(key=mllog_constants.SEED,value=args.random_seed)
     mllogger.end(key=mllog_constants.INIT_STOP)

diff --git a/graph_neural_network/train_rgnn_multi_gpu.py b/graph_neural_network/train_rgnn_multi_gpu.py
@@ -314,7 +314,7 @@ def run_training_proc(rank, world_size,
   submission_info(mllogger, 'GNN', 'reference_implementation')
   mllogger.event(key=mllog_constants.GLOBAL_BATCH_SIZE, value=world_size*args.train_batch_size)
   mllogger.event(key=mllog_constants.GRADIENT_ACCUMULATION_STEPS, value=1)
-  mllogger.event(key=mllog_constants.OPT_NAME, value='Adam')
+  mllogger.event(key=mllog_constants.OPT_NAME, value='adam')
   mllogger.event(key=mllog_constants.OPT_BASE_LR, value=args.learning_rate)
   mllogger.event(key=mllog_constants.SEED,value=args.random_seed)
   mllogger.end(key=mllog_constants.INIT_STOP)