Util classes

simple dataclasses for dataflow management

source

InnerConfig

 InnerConfig (test_train_split:float, input_dim:int, output_dim:int,
              hidden_layer_sizes:list, batch_size:int, epochs:int,
              lr:float, mu:float, n_fns:int, l2_reg:float, seed:int,
              base_act:str='sin')

source

OuterConfig

 OuterConfig (input_dim:int, output_dim:int, hidden_layer_sizes:list,
              batch_size:int, steps:int, print_every:int, lr:float,
              mu:float, seed:int)

source

InnerResults

 InnerResults (train_loss:list, test_loss:list, grad_norm:list)

dataclass to store inner loop results


source

OuterResults

 OuterResults (inner_test_loss:list, train_loss:list, inner_afuncs:list,
               grad_norm:list)

dataclass to store outer loop results