geowatch.tasks.invariants.segmentation_model module

class geowatch.tasks.invariants.segmentation_model.segmentation_model(hparams)[source]

Bases: LightningModule

forward(x, positions=None)[source]
shared_step(batch)[source]
training_step(batch, batch_idx)[source]
validation_step(batch, batch_idx)[source]
test_step(batch)[source]
train_epoch_end(outputs)[source]
validation_epoch_end(outputs)[source]
run_test(loader)[source]
train_dataloader()[source]
val_dataloader()[source]
configure_optimizers()[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.