Merge pull request #4 from Forbu/master

pytorch 0.3 correction
ikostrikov · Feb 9, 2018 · 5faa25b · 5faa25b
2 parents 3824f0b + 7704c48
commit 5faa25b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/meta_optimizer.py b/meta_optimizer.py
@@ -137,7 +137,8 @@ def meta_update(self, model_with_grads, loss):
         self.f, self.i = self(inputs)
 
         # Meta update itself
-        flat_params = self.f * flat_params - self.i * Variable(flat_grads)
+        flat_params = torch.t(self.f) * flat_params - torch.t(self.i) * Variable(flat_grads)
+        flat_params = flat_params.view(-1)
 
         self.meta_model.set_flat_params(flat_params)