{ "config": { "hidden_size": 16, "train_epochs": 3, "inter_period": 100, "inter_epochs": 0, "pre_epochs": 2, "max_inter_episodes": 0, "batch_size": 4, "input_dim": 1, "train_steps": 40, "lyap_steps": 30, "floss_input_steps": 40, "seed_ic": 1, "seed_input": 1, "seed_net": 1, "seed_ons": 1, "lr": 0.001, "beta1": 0.9, "beta2": 0.999, "init_type": 1, "recurrent_gain": 1.0, "recurrent_mean_gain": 0.0, "input_scale": 1.0, "delay": 10, "ws_std": 1.0, "ws_mean": 0.0, "wr_std": 1.0, "wr_mean": 0.0, "b_std": 0.1, "b_mean": 0.0, "n_lyap": 4, "task": -1, "lyap_target": 0.0, "eval_every": 1, "eval_batches": 1, "log_every_floss": 50, "device": "cuda", "out": "/home/yurenh2/rrm/research/flossing/flossing_suite/results/smoke/toy_smoke.json" }, "evals": [ { "epoch": 1, "train_loss": 0.8864837288856506, "train_accuracy": 0.4880952380952381, "eval_loss": 0.8243135213851929, "eval_accuracy": 0.5476190476190477, "elapsed": 0.8400521278381348 }, { "epoch": 2, "train_loss": 0.9001060128211975, "train_accuracy": 0.4880952380952381, "eval_loss": 0.9111667275428772, "eval_accuracy": 0.5238095238095238, "elapsed": 0.8638095855712891 }, { "epoch": 3, "train_loss": 0.9251764416694641, "train_accuracy": 0.4880952380952381, "eval_loss": 0.7443239688873291, "eval_accuracy": 0.6190476190476191, "elapsed": 0.8863844871520996 } ], "floss": [ { "epoch": 1, "kind": "pre", "floss_step": 1, "loss": 0.18372014164924622, "lambda_mean": -0.4003114700317383, "lambda_1": -0.22871017456054688, "elapsed": 0.7358992099761963 }, { "epoch": 1, "kind": "pre", "floss_step": 2, "loss": 0.3812721371650696, "lambda_mean": -0.5845165252685547, "lambda_1": -0.29521623253822327, "elapsed": 0.7633428573608398 } ], "task": [] }