Superxixixi commited on
Commit
6f6705b
·
1 Parent(s): 158e809

Upload loconet

Browse files
Files changed (2) hide show
  1. modeling_loconet.py +3 -0
  2. pytorch_model.bin +2 -2
modeling_loconet.py CHANGED
@@ -11,6 +11,9 @@ class loconet(PreTrainedModel):
11
  super().__init__(config)
12
 
13
  self.model = locoencoder(config)
 
 
 
14
 
15
  def forward(self, audioFeature, visualFeature, masks, labels=None):
16
  b, s, t = visualFeature.shape[:3]
 
11
  super().__init__(config)
12
 
13
  self.model = locoencoder(config)
14
+ self.lossAV = lossAV()
15
+ self.lossA = lossA()
16
+ self.lossV = lossV()
17
 
18
  def forward(self, audioFeature, visualFeature, masks, labels=None):
19
  b, s, t = visualFeature.shape[:3]
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6918e8391d48c40cfd90b332687508bb4b2269879ba1303dacb5a26937ecda87
3
- size 137464429
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e7aa6a388e424747d5b0b204b383ec53cc466a37b7f1c4c7e34a9fc766ce11f2
3
+ size 137470241