geowatch.tasks.invariants.pretext_model module

class geowatch.tasks.invariants.pretext_model.pretext(hparams)[source]

Bases: LightningModule

TASK_NAMES = ['sort', 'augment', 'overlap']
forward(image_stack, positional_encoding=None)[source]
shared_step(batch)[source]
training_step(batch, batch_idx)[source]
validation_step(batch, batch_idx)[source]
predict(batch)[source]
train_dataloader()[source]
val_dataloader()[source]
configure_optimizers()[source]
task_neck(in_chan, out_chan)[source]
pixel_classification_head(in_chan, num_classes)[source]
image_classification_head(in_chan)[source]
generate_pca_matrix(save_path, loader, reduction_dim=6)[source]
on_save_checkpoint(checkpoint)[source]
save_package(package_path)[source]
classmethod load_package(package_path)[source]

DEPRECATE IN FAVOR OF geowatch.tasks.fusion.utils.load_model_from_package

Todo

  • [ ] Make the logic that defines the save_package and load_package

    methods with appropriate package header data a lightning abstraction.