File size: 536 Bytes
29c9ba5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
from transformers import PretrainedConfig
class SSLConfig(PretrainedConfig):
model_type = "ssl-aasist"
def __init__(
self,
filts = [128, [1, 32], [32, 32], [32, 64], [64, 64]],
gat_dims = [64, 32],
pool_ratios = [0.5, 0.5, 0.5, 0.5],
temperatures = [2.0, 2.0, 100.0, 100.0],
**kwargs,
):
self.filts = filts
self.gat_dims = gat_dims
self.pool_ratios = pool_ratios
self.temperatures = temperatures
super().__init__(**kwargs) |