File size: 731 Bytes
983931b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
"""Configuration classes for MoCo models compatible with transformers"""
from transformers import PretrainedConfig
class MoCoResNetConfig(PretrainedConfig):
"""Configuration for MoCo ResNet models"""
model_type = "moco_resnet"
def __init__(
self,
arch="resnet50",
block="Bottleneck",
layers=[3, 4, 6, 3],
image_size=224,
num_channels=3,
num_labels=0,
epoch=0,
**kwargs
):
super().__init__(**kwargs)
self.arch = arch
self.block = block
self.layers = layers
self.image_size = image_size
self.num_channels = num_channels
self.num_labels = num_labels
self.epoch = epoch
|