Merge pull request #548 from xinyuewang1/patch-2

Update train_a3c.py
chainer · Sep 4, 2019 · ef2f21e · ef2f21e
2 parents c9284a4 + e4c8e02
commit ef2f21e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/atari/reproduction/a3c/train_a3c.py b/examples/atari/reproduction/a3c/train_a3c.py
@@ -23,7 +23,7 @@
 from chainerrl.optimizers.nonbias_weight_decay import NonbiasWeightDecay
 from chainerrl.optimizers import rmsprop_async
 from chainerrl import policies
-from chainerrl import v_function
+from chainerrl import v_functions
 
 from chainerrl.wrappers import atari_wrappers
 
@@ -34,7 +34,7 @@ def __init__(self, n_actions):
         self.head = links.NIPSDQNHead()
         self.pi = policies.FCSoftmaxPolicy(
             self.head.n_output_channels, n_actions)
-        self.v = v_function.FCVFunction(self.head.n_output_channels)
+        self.v = v_functions.FCVFunction(self.head.n_output_channels)
         super().__init__(self.head, self.pi, self.v)
 
     def pi_and_v(self, state):