@@ -24,6 +24,8 @@ L2 = "l2"
def model_dot_product(w1, w2, requires_grad=True):
""" Return the sum of squared difference between two models. """
+ print(w1)
+ print(w2)
dot_product = 0.0
for p1, p2 in zip(w1.parameters(), w2.parameters()):
if requires_grad:
