logo
Browse Source

add3

Signed-off-by: xujinling <jinling.xu@zilliz.com>
main
xujinling 3 years ago
parent
commit
4821b6a729
  1. 2
      frozen_in_time.py
  2. BIN
      frozen_in_time_base_16_224.pth

2
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:

BIN
frozen_in_time_base_16_224.pth (Stored with Git LFS)

Binary file not shown.
Loading…
Cancel
Save