keishihara commited on
Commit
ecaafec
·
verified ·
1 Parent(s): 6279c12

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. model.py +0 -2
  2. modeling_act_estimator.py +1 -2
model.py CHANGED
@@ -6,8 +6,6 @@ import torch.nn.functional as F
6
  from torch import Tensor, nn
7
  from torch.nn import TransformerEncoder, TransformerEncoderLayer
8
 
9
- from transformers import AutoModel, PreTrainedModel
10
-
11
 
12
  class MaxPool3dSamePadding(nn.MaxPool3d):
13
  def compute_pad(self, dim, s):
 
6
  from torch import Tensor, nn
7
  from torch.nn import TransformerEncoder, TransformerEncoderLayer
8
 
 
 
9
 
10
  class MaxPool3dSamePadding(nn.MaxPool3d):
11
  def compute_pad(self, dim, s):
modeling_act_estimator.py CHANGED
@@ -1,9 +1,8 @@
1
- import torch
2
  from torch import Tensor
3
  from transformers import PreTrainedModel
4
 
5
- from .model import VideoActionEstimator
6
  from .configuration_act_estimator import ActEstimatorConfig
 
7
 
8
 
9
  class ActEstimator(PreTrainedModel):
 
 
1
  from torch import Tensor
2
  from transformers import PreTrainedModel
3
 
 
4
  from .configuration_act_estimator import ActEstimatorConfig
5
+ from .model import VideoActionEstimator
6
 
7
 
8
  class ActEstimator(PreTrainedModel):