keishihara commited on
Commit
466dc17
·
verified ·
1 Parent(s): 3d729d6

Upload folder using huggingface_hub

Browse files
configuration_act_estimator.py CHANGED
@@ -25,5 +25,3 @@ class ActEstimatorConfig(PretrainedConfig):
25
  self.dropout = dropout
26
  self.feature_map_size = feature_map_size
27
  super().__init__(**kwargs)
28
-
29
-
 
25
  self.dropout = dropout
26
  self.feature_map_size = feature_map_size
27
  super().__init__(**kwargs)
 
 
modeling_act_estimator.py CHANGED
@@ -15,22 +15,3 @@ class ActEstimator(PreTrainedModel):
15
 
16
  def forward(self, frames: Tensor, timestamps: Tensor = None) -> dict[str, Tensor]:
17
  return self.model(frames, timestamps)
18
-
19
-
20
-
21
- # actestimator_config = ActEstimatorConfig.from_pretrained(".")
22
- # print(actestimator_config.to_dict())
23
-
24
- # actestimator_model = ActEstimator(actestimator_config)
25
-
26
- # state_dict = torch.load("ckpt.pth", weights_only=True)
27
- # actestimator_model.model.load_state_dict(state_dict)
28
-
29
- # print(actestimator_model)
30
- # actestimator_model.save_pretrained(".")
31
-
32
-
33
-
34
- # model = ActEstimator.from_pretrained(".")
35
- # print(model(torch.randn(1, 3, 44, 224, 224), torch.randn(1, 44)))
36
-
 
15
 
16
  def forward(self, frames: Tensor, timestamps: Tensor = None) -> dict[str, Tensor]:
17
  return self.model(frames, timestamps)