mv block in train

train.py

@@ -1,5 +1,7 @@
 #!/usr/bin/env python3
 #
+# train.py
+#
 # SPDX-License-Identifier: BSD-2-Clause
 #
 # Copyright (c) 2023, Jeff Moe
@@ -29,19 +31,11 @@
 
 import argparse
 
+# PyTorch version used in paper ?
 # Python version used in paper ?
 # Set hyperparameters based on values in Table 3 of
 # Pl@ntNet-300K paper.
 #
-# Initial Learning Rate
-LR='N.NNNN'
-# Number of Epochs
-N_EPOCHS='NN'
-# First Decay
-FIRST_DECAY='1N'
-# Secon Decay
-SECOND_DECAY='2N'
-
 # Use defaults from git repo example.
 # https://github.com/plantnet/PlantNet-300K
 BATCH_SIZE='32'
@@ -73,6 +67,14 @@ parser.add_argument('model',
 args = parser.parse_args()
 MODEL_NAME = args.model
 
+# Initial Learning Rate
+LR='N.NNNN'
+# Number of Epochs
+N_EPOCHS='NN'
+# First Decay
+FIRST_DECAY='1N'
+# Secon Decay
+SECOND_DECAY='2N'
 # Set LR, epochs, decay hyperparameters based on model.
 match MODEL_NAME:
   case "alexnet":