File size: 300 Bytes
8b78124 | 1 2 3 4 5 6 7 8 9 10 11 | from transformers import PretrainedConfig
class PredNetConfig(PretrainedConfig):
model_type = "prednet"
def __init__(self, sequence_len=4, resize_hw=(128, 128), **kwargs):
super().__init__(**kwargs)
self.sequence_len = sequence_len
self.resize_hw = list(resize_hw)
|