ride.finetune

Module Contents

Classes

Finetunable

Adds finetune capabilities to model

Functions

load_model_weights(file, hparams_passed, model_state_key)
try_pyth_load(file, model_state_key)
try_pickle_load(file)

Attributes

logger

ride.finetune.logger

class ride.finetune.Finetunable(hparams: pytorch_lightning.utilities.parsing.AttributeDict, *args, **kwargs)

Bases: ride.unfreeze.Unfreezable

Adds finetune capabilities to model

This module is automatically added when RideModule is inherited

hparams: Ellipsis

static configs() → ride.core.Configs

validate_attributes()

map_loaded_weights(file, loaded_state_dict)

on_init_end(hparams, *args, **kwargs)

ride.finetune.load_model_weights(file: str, hparams_passed, model_state_key)

ride.finetune.try_pyth_load(file, model_state_key)

ride.finetune.try_pickle_load(file)