geowatch.tasks.invariants.late_fusion.pretext_model_late_fusion module

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

Bases: LightningModule

TASK_NAMES = ['sort', 'augment', 'overlap']
forward(image)[source]
shared_step(batch)[source]
training_step(batch, batch_idx)[source]
validation_step(batch, batch_idx)[source]
predict(batch)[source]
predict_before_after(image)[source]
train_dataloader()[source]
val_dataloader()[source]
configure_optimizers()[source]
task_neck(in_chan, out_chan)[source]
pixel_classification_head(in_chan)[source]
image_classification_head(in_chan)[source]