DetailNet / config.json
tsonubin's picture
DetailNet r2: weights, config and model card
8b065a3 verified
Raw History Blame Contribute Delete
542 Bytes
{
"architecture": "DetailNet",
"release": "r2",
"scale": 3,
"channels": 64,
"blocks": 12,
"hidden": 32,
"parameters": 972795,
"training": {
"iters": 14000,
"batch": 4,
"lr": 0.0002,
"weight_decay": 0.0001,
"w_perc": 1.0,
"w_l1": 1.0,
"w_lock": 1.0,
"w_pp": 0.5,
"seed": 1,
"crop": [
160,
90
],
"reset_prob": 0.15,
"pp_prob": 0.2,
"perc_frames": 4,
"grad_clip": 1.0,
"initialised_from": "r1 (iteration 18000)",
"device": "Apple M5 Max (MPS)"
}
}