phenocam_snow.model

class PhenoCamResNet(resnet, n_classes, lr=0.0005, weight_decay=0.01)

Loads pre-trained ResNet for fine-tuning.

__init__(resnet, n_classes, lr=0.0005, weight_decay=0.01)

Parameters:

• resnet (str) – The ResNet variant to use.

• n_classes (int) – The number of classes

• lr (float) – The learning rate. Default is 1e-5.

• weight_decay (float) – The weight decay to use. Default is 0.1.