set max-iter to inf

Author: scientist1642

main.py

@@ -3,6 +3,7 @@
 import argparse
 import os
 import sys
+import math
 
 import torch
 import torch.optim as optim
@@ -40,7 +41,7 @@
                     help='environment to train on (default: PongDeterministic-v3)')
 parser.add_argument('--no-shared', default=False, metavar='O',
                     help='use an optimizer without shared momentum.')
-parser.add_argument('--max-iters', type=int, default=50000,
+parser.add_argument('--max-iters', type=int, default=math.inf,
                     help='maximum iterations per process.')
 
 parser.add_argument('--debug', action='store_true', default=False,