diff --git a/main.py b/main.py index 4df884d..ffe14ba 100644 --- a/main.py +++ b/main.py @@ -385,7 +385,7 @@ if __name__ == '__main__': # ppo_cart_pole() # dqn_pixel_atari('BreakoutNoFrameskip-v4') - a2c_pixel_atari('BreakoutNoFrameskip-v4') + # a2c_pixel_atari('BreakoutNoFrameskip-v4') # categorical_dqn_pixel_atari('BreakoutNoFrameskip-v4') # quantile_regression_dqn_pixel_atari('BreakoutNoFrameskip-v4') # n_step_dqn_pixel_atari('BreakoutNoFrameskip-v4') diff --git a/utils/normalizer.py b/utils/normalizer.py index cfb65ff..dbaaee1 100644 --- a/utils/normalizer.py +++ b/utils/normalizer.py @@ -78,7 +78,7 @@ class RunningStatsNormalizer(BaseNormalizer): class RescaleNormalizer(BaseNormalizer): def __init__(self, coef=1.0): - super(RescaleNormalizer, self).__init__() + BaseNormalizer.__init__(self) self.coef = coef def __call__(self, x):