geowatch.tasks.depth.pl_highres_verify module

geowatch.tasks.depth.pl_highres_verify.modify_bn(model, track_running_stats=True, bn_momentum=0.1)

class geowatch.tasks.depth.pl_highres_verify.MultiTaskModel(batch_size: int = 1, checkpoint: str = None, config: dict = None, test_img_dir: str = None, test_img_list: str = None, gpus: str = '0', **kwargs)

Bases: LightningModule

forward(x, tta=False)

test_step(batch, batch_idx)

static add_model_specific_args(parent_parser)