reni-models / minimal /decoder_config.json
jadgardner's picture
Release RENI Models v1.1 with minimal PyTorch decoder
25b7b90 verified
Raw
History Blame Contribute Delete
860 Bytes
{
"format_version": 1,
"model_type": "reni-vnjoint-ortho-so2-two-bracket",
"config": {
"latent_dim": 100,
"axis_of_invariance": 2,
"hidden_features": 128,
"num_attention_heads": 8,
"num_attention_layers": 6,
"out_features": 6,
"num_frequencies": 2,
"min_frequency_exponent": 0.0,
"max_frequency_exponent": 2.0,
"include_direction_input": true,
"m_ldr": 16.0,
"m_log": 10000.0,
"blend_tau": 0.95,
"blend_delta": 0.02
},
"source": {
"model_id": "thesis-vnjoint-ortho-so2-d100",
"checkpoint": "step-000100001.ckpt",
"checkpoint_sha256": "40a3806306304cb51cc64424f9a27950c9283ddd470d3df12c995b49a043472f",
"step": 100001
},
"artifact": {
"path": "decoder.pt",
"size_bytes": 3744305,
"sha256": "aeed4447a691447e645b96f7269f45c0097570b93da35bad416cb4bb14de3989"
}
}