Hi, Thank you for providing the quality code. I have a implementation detail question.
- Why don't you just call optimizer.step() after meta_objective.backward()?
|
self._meta_optimizer.zero_grad() |
|
meta_objective.backward() |
|
|
|
self._meta_optimize(all_samples, all_params) |
Hi, Thank you for providing the quality code. I have a implementation detail question.
garage/src/garage/torch/algos/maml.py
Lines 124 to 127 in f056fb8