anypinn.cli.scaffold.lorenz.config

Lorenz system — training configuration.

EXPERIMENT_NAME = '__EXPERIMENT_NAME__' module-attribute

RUN_NAME = 'v0' module-attribute

hp = ODEHyperparameters(lr=0.001, max_epochs=3000, gradient_clip_val=0.5, criterion='huber', training_data=_training_data, fields_config=(MLPConfig(in_dim=1, out_dim=1, hidden_layers=[64, 128, 128, 64], activation='tanh', output_activation=None)), params_config=(ScalarConfig(init_value=15.0)), scheduler=(ReduceLROnPlateauConfig(mode='min', factor=0.5, patience=55, threshold=0.005, min_lr=1e-06)), pde_weight=0.001, ic_weight=10, data_weight=5) module-attribute