Upload EVPDepth

model.py

@@ -327,7 +327,7 @@ class EVPDepthEncoder(nn.Module):
             param.requires_grad = True
         
         self.text_adapter = TextAdapterRefer(text_dim=text_dim)
-        self.gamma = nn.Parameter(torch.ones(text_dim) * 1e-4)
+        self.alpha = nn.Parameter(torch.ones(text_dim) * 1e-4)
         device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
         if caption_aggregation:                
@@ -398,7 +398,7 @@ class EVPDepthEncoder(nn.Module):
         else:
             class_embeddings = self.class_embeddings
         
-        c_crossattn = self.text_adapter(latents, class_embeddings, self.gamma)
+        c_crossattn = self.text_adapter(latents, class_embeddings, self.alpha)
         t = torch.ones((x.shape[0],), device=x.device).long()
 
         #if self.dataset == 'kitti':

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b67d266d40074d28fd30fe8b341ce964d80773ad4bb0eb7d321fc4454e9c461b
+oid sha256:83540a8d86d764c873c5d7e032d4c753751bde4165021cfe78b4478aa983af33
 size 3735516436