Index

_

__getitem__() (base_loader.ModelDataset method)

__len__() (base_loader.ModelDataset method)

A

add_bias() (base_model.BaseModel method)

apply_mask() (base_loader.ModelDataset method) autolabel() (in module test)

B

base_loader module base_model module

BaseModel (class in base_model) benchmark_step() (base_loader.ModelDataset method)

C

config module

configure_optimizers() (base_model.BaseModel method) create_config() (in module config)

F

forward() (base_model.BaseModel method) (unet.Model method) (unet_interpolated.Model method) (unet_snipped.Model method) (unet_tapered.Model method)

fwi_forecast module fwi_reanalysis module

G

generate_isolated_mask() (gfas_frp.ModelDataset method) get_cb_loss_factor() (base_loader.ModelDataset method) get_hparams() (in module config) (in module plot)

get_loss() (base_loader.ModelDataset method) get_model() (in module train) gfas_frp module

I

inference_step() (base_loader.ModelDataset method)

M

main() (in module test) (in module train) make_dict() (in module plot) Model (class in unet) (class in unet_downsampled) (class in unet_interpolated) (class in unet_snipped) (class in unet_tapered) ModelDataset (class in base_loader) (class in fwi_forecast) (class in fwi_reanalysis) (class in gfas_frp) module base_loader base_model config fwi_forecast fwi_reanalysis gfas_frp plot test train unet unet_downsampled unet_interpolated unet_snipped unet_tapered

multi_day_plot() (in module plot)

P

plot module

plot() (in module plot) prepare_data() (base_model.BaseModel method)

R

read_json() (in module plot)

run() (in module test)

S

set_hparams() (in module train) single_day_plot() (in module plot) (in module test)

str2num() (in module config) (in module plot)

T

test module test_dataloader() (base_model.BaseModel method) test_epoch_end() (base_model.BaseModel method) (unet_interpolated.Model method) test_step() (base_loader.ModelDataset method) (base_model.BaseModel method) (gfas_frp.ModelDataset method)

train module train_dataloader() (base_model.BaseModel method) training_epoch_end() (base_model.BaseModel method) training_step() (base_loader.ModelDataset method) (base_model.BaseModel method) (gfas_frp.ModelDataset method)

U

unet module unet_downsampled module unet_interpolated module

unet_snipped module unet_tapered module

V

val_dataloader() (base_model.BaseModel method) validation_epoch_end() (base_model.BaseModel method)

validation_step() (base_loader.ModelDataset method) (base_model.BaseModel method) (gfas_frp.ModelDataset method)