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