Update main.py

KusanagiNaoya · Jun 8, 2024 · f887190 · f887190
1 parent c1406db
commit f887190
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/2.1_Duel-Double-DQN/main.py b/2.1_Duel-Double-DQN/main.py
@@ -14,7 +14,7 @@
 parser.add_argument('--write', type=str2bool, default=False, help='Use SummaryWriter to record the training')
 parser.add_argument('--render', type=str2bool, default=False, help='Render or Not')
 parser.add_argument('--Loadmodel', type=str2bool, default=False, help='Load pretrained model or Not')
-parser.add_argument('--ModelIdex', type=int, default=100*1000, help='which model to load')
+parser.add_argument('--ModelIdex', type=int, default=100, help='which model to load')
 
 parser.add_argument('--seed', type=int, default=0, help='random seed')
 parser.add_argument('--Max_train_steps', type=int, default=int(1e6), help='Max training steps')