PIVOT / results /validation /training.json
pranamanam's picture
Upload 176 files
6fa9282 verified
Raw History Blame Contribute Delete
1.16 kB
{
"protocol": "split-first-v1",
"config": {
"d_pert": 16,
"hidden": 64,
"depth": 2,
"epochs": 2,
"batch_size": 64,
"lr": 0.001,
"weight_decay": 1e-05,
"lam_tan": 1.0,
"lam_semi": 0.5,
"lam_reg": 0.0001,
"lam_dist": 0.0,
"n_dist_perts": 4,
"dist_n": 64,
"grad_clip": 5.0,
"match": "random",
"rep_mode": "gene_op",
"seed": 0,
"device": "cpu",
"threads": 4,
"endpoint_weight": 0.0
},
"history": [
{
"epoch": 0,
"validation_mse": 0.15263483114540577,
"map": 3.9591146111488342,
"tan": 24.06910252571106,
"semi": 2.5752618526553306e-06,
"reg": 0.010355364764109254,
"total": 28.028218746185303
},
{
"epoch": 1,
"validation_mse": 0.14963708445429802,
"map": 3.667565017938614,
"tan": 23.406102418899536,
"semi": 1.3794184610560478e-05,
"reg": 0.01302928978111595,
"total": 27.073675632476807
}
],
"duration_s": 0.0614291250021779,
"n_train_cells": 504,
"n_parameters": 8602,
"software": {
"python": "3.12.14",
"torch": "2.14.0+cpu",
"numpy": "2.3.5"
}
}