Try standard params
@@ -175,8 +175,6 @@ def calc_loss(batch, net, tgt_net, gamma, cuda=False, save_prefix=None):
if __name__ == "__main__":
params = common.HYPERPARAMS['pong']
- params['epsilon_frames'] = 200000
- params['replay_size'] = 200000
parser = argparse.ArgumentParser()
parser.add_argument("--cuda", default=False, action="store_true", help="Enable cuda")
args = parser.parse_args()
