Parallax-LM's picture
Update config.json
77c9adc verified
raw
history blame contribute delete
645 Bytes
{
"architectures": [
"ParallaxVisionAutoencoder"
],
"model_type": "parallax_vision",
"model_name": "parallax_VISION-boxes-RGBA",
"total_parameters": 493876,
"input_resolution": [720, 720],
"channels": 4,
"latent_dims": [45, 45, 4],
"compression_ratio": 256.0,
"torch_dtype": "float32",
"_name_or_path": "Parallax-labs-1/parallax_VISION-boxes-RGBA",
"model_file": "model.pt",
"training_parameters": {
"steps": 10000,
"optimizer": "Adam",
"precision": "float32",
"final_loss": 0.000038
},
"dataset": {
"type": "procedural",
"source": "box-generator-v2",
"domain": "RGBA-Squares"
}
}