There was an error while loading. Please reload this page.
1 parent 9521e56 commit 46b6af5Copy full SHA for 46b6af5
reptile/main.py
@@ -129,6 +129,7 @@ def REPTILE(
129
epochs: int = EPOCHS,
130
) -> Weights:
131
"""Run one iteration of REPTILE."""
132
+
133
weights = [sgd(meta_weights, epochs) for _ in range(meta_batch_size)]
134
135
# TODO Implement custom optimizer that makes this work with builtin
@@ -187,7 +188,7 @@ def REPTILE(
187
188
if PLOT:
189
190
ax.set_title(f'REPTILE after {iteration:n} iterations')
- curve, = ax.plot(
191
+ (curve, ) = ax.plot(
192
x_all.numpy(),
193
model(Variable(x_all)).data.numpy(),
194
label=f'Pred after {TEST_GRAD_STEPS} gradient steps.',
0 commit comments