From 9bda6d9b7919a75e55978eb0e5be56f18a4813d6 Mon Sep 17 00:00:00 2001 From: Themaoqiu <2873133716@qq.com> Date: Thu, 5 Dec 2024 23:40:00 +0800 Subject: [PATCH] Update main.py --- main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index 65e0d9e..9ba1e11 100644 --- a/main.py +++ b/main.py @@ -15,7 +15,7 @@ import helpers.data_helpers as dh import methods.clip_transformations as CLIPTransformations from utils import read_unknowns, nest_dict -from clip_utils import get_features, evaluate, zeroshot_classifier, get_ensamble_preds, get_pred_overlap, get_nn_metrics +from clip_utils import get_features, evaluate, zeroshot_classifier, get_ensamble_preds, get_pred_overlap, get_nn_metrics, load_embeddings import methods.augmentations parser = argparse.ArgumentParser(description='CLIP Advice') @@ -238,4 +238,4 @@ def flatten_config(dic, running_key=None, flattened_dict={}): axs_new[i].set_title(f"{dataset_domains[int(old_test_domains[int(sample_idx)])]} - {old_test_labels[int(sample_idx)]}") axs_new[i].axis('off') wandb.log({"train features NN": wandb.Image(f), "domain consistency acc": domain_acc, "class consistency acc": class_acc, "unique nn": prop_unique}) - wandb.sklearn.plot_confusion_matrix(sample_domains, neighbor_domains, dataset_domains) \ No newline at end of file + wandb.sklearn.plot_confusion_matrix(sample_domains, neighbor_domains, dataset_domains)