def load_pretrained(self, encoder_name):
available_weights = {"resnet18"}
if encoder_name in available_weights:
load_pretrained(self, f"CPC_v2-{encoder_name}")
elif encoder_name not in available_weights:
rank_zero_warn(f"{encoder_name} not yet available")
def load_pretrained(self, encoder_name):
available_weights = {'resnet18'}
if encoder_name in available_weights:
load_pretrained(self, f'CPCV2-{encoder_name}')
elif encoder_name not in available_weights:
rank_zero_warn(f'{encoder_name} not yet available')
def load_pretrained(self, pretrained):
available_weights = {'imagenet2012'}
if pretrained in available_weights:
weights_name = f'vae-{pretrained}'
load_pretrained(self, weights_name)
def load_pretrained(self, pretrained):
available_weights = {"imagenet2012"}
if pretrained in available_weights:
weights_name = f"vae-{pretrained}"
load_pretrained(self, weights_name)
