From a32bc9c828df054a23802d96eaf030656f37c748 Mon Sep 17 00:00:00 2001 From: Andrew Zhao Date: Wed, 2 Dec 2020 11:09:43 +0800 Subject: [PATCH] bug fixes --- simsiam/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/simsiam/main.py b/simsiam/main.py index ab76030..d0bd964 100644 --- a/simsiam/main.py +++ b/simsiam/main.py @@ -122,7 +122,7 @@ def cosine_loss(p, z): linear_classifier.fit(feature_bank, feature_labels) y_preds, y_trues = [], [] - for data, target in feature_loader: + for data, target in test_loader: data = data.to(device) with torch.no_grad(): feature = model(data, istrain=False)