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)