add3

Signed-off-by: xujinling <jinling.xu@zilliz.com>
3 years ago · 4821b6a729
2 changed files with 0 additions and 5 deletions
--- a/frozen_in_time.py
+++ b/frozen_in_time.py
@ -50,8 +50,6 @@ class FrozenInTime(NNOperator):
        super().__init__()
        self.model_name = model_name
        self.modality = modality
        # if weight_path is None:
        #     weight_path = str(Path(__file__).parent / 'frozen_in_time_base_16_224.pth')
        if device is None:
            self.device = "cuda" if torch.cuda.is_available() else "cpu"
        else:
--- a/frozen_in_time_base_16_224.pth
+++ b/frozen_in_time_base_16_224.pth