File size: 550 Bytes
679a6b4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
from transformers import PretrainedConfig
class ReVarConfig(PretrainedConfig):
model_type = "revar"
def __init__(self, outer_dim: int = 960, inner_dim: int = 480, kernel_size: int = 5, stack_size: int = 2, num_stacks: int = 20, num_output_channels: int = 5, **kwargs):
self.outer_dim = outer_dim
self.inner_dim = inner_dim
self.kernel_size = kernel_size
self.stack_size = stack_size
self.num_stacks = num_stacks
self.num_output_channels= num_output_channels
super().__init__(**kwargs) |