template_classifier / configuration_cetacean_classifier.py
vancauwe's picture
feat: template for model on hugging face
fe3b346 verified
raw
history blame contribute delete
225 Bytes
from transformers import PretrainedConfig
class TemplateClassifierConfig(PretrainedConfig):
model_type = "TEMPLATE_MODEL_NAME"
def __init__(
self,
**kwargs,
):
super().__init__(**kwargs)