Commit 847c07f1 authored by Nicola Gatto's avatar Nicola Gatto

Fix wrong tests

parent fc007f99
Pipeline #163026 failed with stages
......@@ -3,7 +3,7 @@ package mountaincar.agent;
component MountaincarCritic {
ports
in Q^{2} state,
in Q^{2} action,
in Q(-1:1)^{1} action,
out Q(-oo:oo)^{1} qvalues;
implementation CNN {
......
......@@ -50,7 +50,7 @@ class CNNCreator_mountaincar_agent_mountaincarCritic:
self.networks[0] = Net_0(data_mean=data_mean, data_std=data_std)
self.networks[0].collect_params().initialize(self.weight_initializer, ctx=context)
self.networks[0].hybridize()
self.networks[0](mx.nd.zeros((1, 2,), ctx=context), mx.nd.zeros((1, 2,), ctx=context))
self.networks[0](mx.nd.zeros((1, 2,), ctx=context), mx.nd.zeros((1, 1,), ctx=context))
if not os.path.exists(self._model_dir_):
os.makedirs(self._model_dir_)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment