You can deploy this model and start earning money today!

pszmk/
mnist-vae-latent2

Name: mnist-vae-latent2
Availability: InStock

Views

No views yet

MNIST VAE latent2

Overview

This model was exported from the Lightning checkpoint epoch=31-step=15008-val_loss=0.2625.ckpt.

Loading

python

from transformers import AutoModel

model = AutoModel.from_pretrained("pszmk/mnist-vae-latent2", trust_remote_code=True)

Load a specific release with a revision or tag:

python

from transformers import AutoModel

model = AutoModel.from_pretrained(
    "pszmk/mnist-vae-latent2",
    revision="main",
    trust_remote_code=True,
)

Provenance

Source checkpoint: /home/pszmk/Latent-Anti-Microbial-Peptides-LAMP/checkpoints/6b865455ff1f4cb7818d40a26f0eeeed/epoch=31-step=15008-val_loss=0.2625.ckpt
Model class: modelling.src.models.mnist_vae.MNISTVAE
Config class: modelling.src.models.mnist_vae.MNISTVAEConfig
Suggested revision: main

Metadata

{}