| { | |
| "architectures": [ | |
| "GlobalHCA_AE", | |
| "LatentPredictor" | |
| ], | |
| "model_type": "encoder-decoder", | |
| "input_size": { | |
| "channels": 4, | |
| "height": 45, | |
| "width": 45 | |
| }, | |
| "latent_dim": 1012, | |
| "hidden_dims": [2048, 4096], | |
| "activation": "relu", | |
| "output_activation": "sigmoid", | |
| "files": [ | |
| { | |
| "file": "ae_global.pt", | |
| "type": "Autoencoder weights" | |
| }, | |
| { | |
| "file": "predictor.pt", | |
| "type": "Latent predictor weights" | |
| }, | |
| { | |
| "file": "model.pt", | |
| "type": "Vision base weights" | |
| } | |
| ], | |
| "version": "1.0.0" | |
| } | |