Update train_mlp_batches.py

train_mlp_batches.py

@@ -81,7 +81,7 @@ def main():
     parser.add_argument('--width', type=int, default=512, help='Number of neurons per hidden layer (default: 512)')
     parser.add_argument('--batch_size', type=int, default=8, help='Batch size for training (default: 8)')
     parser.add_argument('--save_model_dir', type=str, default='saved_models', help='Directory to save model checkpoints (default: saved_models)')
-    parser.add_argument('--access_token', type=str, required=True, help='ModelScope SDK access token')
+    parser.add_argument('--access_token', type=str, help='ModelScope SDK access token (optional)')
     parser.add_argument('--upload_checkpoint', action='store_true', help='Upload checkpoint to ModelScope')
     parser.add_argument('--delete_checkpoint', action='store_true', help='Delete local checkpoint after uploading')
     args = parser.parse_args()
@@ -163,6 +163,8 @@ def main():
 
     # Upload the model to ModelScope if specified
     if args.upload_checkpoint:
+        if not args.access_token:
+            raise ValueError("Access token is required for uploading to ModelScope.")
         api = HubApi()
         api.login(args.access_token)
         api.push_model(