CatkinChen
/

nethack-vae-hmm

Feature Extraction

MultiModalHackVAE_with_StickyHDPHMM

reinforcement-learning

variational-autoencoder

representation-learning

Model card Files Files and versions

nethack-vae-hmm

42.2 MB

1 contributor

History: 129 commits

CatkinChen's picture

Upload VAE+HMM round 1/3 (train_loss=3936.1468)

4642014 verified 3 months ago

training_plots
Add training curves 3 months ago
.gitattributes

1.59 kB

Add training curves 4 months ago
README.md

2.06 kB

Add model card 4 months ago
config.json

692 Bytes

Add model config 3 months ago
pytorch_model.bin
Detected Pickle imports (5)
- "torch._utils._rebuild_tensor_v2",
- "src.model.VAEConfig",
- "torch.FloatStorage",
- "torch.LongStorage",
- "collections.OrderedDict"
How to fix it?
9.36 MB
xet

Upload VAE+HMM round 1/3 (train_loss=3936.1468) 3 months ago
training_data.json

1.36 kB

Add training data 3 months ago
vae_with_hmm_round1.pt

10.9 MB
xet

Add vae_with_hmm_round1.pt 3 months ago
vae_with_hmm_round2.pt
Detected Pickle imports (10)
- "collections.OrderedDict",
- "torch.device",
- "torch.FloatStorage",
- "src.skill_space.StickyHDPHMMParams",
- "torch.LongStorage",
- "src.skill_space.NIWPrior",
- "torch.float32",
- "torch._utils._rebuild_tensor_v2",
- "torch.DoubleStorage",
- "src.model.VAEConfig"
How to fix it?
10.9 MB
xet

Add vae_with_hmm_round2.pt 3 months ago
vae_with_hmm_round3.pt
Detected Pickle imports (10)
- "torch._utils._rebuild_tensor_v2",
- "src.skill_space.StickyHDPHMMParams",
- "torch.FloatStorage",
- "torch.LongStorage",
- "collections.OrderedDict",
- "torch.DoubleStorage",
- "src.skill_space.NIWPrior",
- "src.model.VAEConfig",
- "torch.float32",
- "torch.device"
How to fix it?
10.9 MB
xet

Add vae_with_hmm_round3.pt 3 months ago