File size: 721 Bytes
3d729d6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
from transformers import PretrainedConfig
class ActEstimatorConfig(PretrainedConfig):
model_type = "ACT-Estimator"
def __init__(
self,
input_shape=(3, 44, 224, 224),
num_classes=9,
max_seq_len=44,
timestamp_dim=1,
d_model=512,
num_heads=8,
dropout=0.1,
feature_map_size=4,
**kwargs
):
self.input_shape = input_shape
self.num_classes = num_classes
self.max_seq_len = max_seq_len
self.timestamp_dim = timestamp_dim
self.d_model = d_model
self.num_heads = num_heads
self.dropout = dropout
self.feature_map_size = feature_map_size
super().__init__(**kwargs)
|