anypinn.cli.scaffold.sir.config_synthetic

SIR epidemic model — training configuration.

EXPERIMENT_NAME = '__EXPERIMENT_NAME__' module-attribute

RUN_NAME = 'v0' module-attribute

hp = ODEHyperparameters(lr=0.0005, max_epochs=2000, gradient_clip_val=0.1, training_data=(GenerationConfig(batch_size=100, data_ratio=2, collocations=6000, x=(torch.linspace(start=0, end=90, steps=91)), args_to_train={}, noise_level=0)), fields_config=(MLPConfig(in_dim=1, out_dim=1, hidden_layers=[64, 128, 128, 64], activation='tanh', output_activation='softplus')), params_config=(MLPConfig(in_dim=1, out_dim=1, hidden_layers=[64, 128, 128, 64], activation='tanh', output_activation='softplus')), scheduler=(ReduceLROnPlateauConfig(mode='min', factor=0.5, patience=55, threshold=0.005, min_lr=1e-06)), pde_weight=1, ic_weight=1, data_weight=1) module-attribute