pszmk/mnist-vae-latent2

MNIST VAE latent2

Overview

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

Loading

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:

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

{}