File size: 291 Bytes
be6e105 |
1 2 3 4 5 6 7 8 9 10 |
from transformers import PretrainedConfig
class CustomResNetConfig(PretrainedConfig):
model_type = "custom-resnet"
def __init__(self, num_labels=2, **kwargs):
super().__init__(**kwargs)
self.num_labels = num_labels # Register number of labels (output dimensions)
|