Hugging Face
Models
Datasets
Spaces
Community
Docs
Enterprise
Pricing
Log In
Sign Up
CatkinChen
/
nethack-vae-hmm
like
0
Feature Extraction
PyTorch
English
MultiModalHackVAE_with_StickyHDPHMM
nethack
reinforcement-learning
variational-autoencoder
representation-learning
multimodal
world-modeling
License:
mit
Model card
Files
Files and versions
xet
Community
eecc510
nethack-vae-hmm
53.2 MB
1 contributor
History:
639 commits
CatkinChen
Add training data
eecc510
verified
3 months ago
training_plots
Add training curves
3 months ago
.gitattributes
Safe
1.59 kB
Add training curves
3 months ago
README.md
Safe
2.06 kB
Add model card
3 months ago
config.json
765 Bytes
Add model config
3 months ago
pytorch_model.bin
pickle
Detected Pickle imports (5)
"torch.FloatStorage"
,
"collections.OrderedDict"
,
"torch.LongStorage"
,
"src.model.VAEConfig"
,
"torch._utils._rebuild_tensor_v2"
How to fix it?
9.36 MB
xet
Upload VAE+HMM round 1/4 (train_loss=3815.7068)
3 months ago
training_data.json
1.44 kB
Add training data
3 months ago
vae_with_hmm_round1.pt
pickle
Detected Pickle imports (10)
"src.model.VAEConfig"
,
"src.skill_space.StickyHDPHMMParams"
,
"collections.OrderedDict"
,
"torch.FloatStorage"
,
"torch._utils._rebuild_tensor_v2"
,
"torch.DoubleStorage"
,
"torch.float32"
,
"torch.LongStorage"
,
"src.skill_space.NIWPrior"
,
"torch.device"
How to fix it?
11 MB
xet
Add vae_with_hmm_round1.pt
3 months ago
vae_with_hmm_round2.pt
pickle
Detected Pickle imports (10)
"torch.float32"
,
"torch.DoubleStorage"
,
"torch.FloatStorage"
,
"src.skill_space.StickyHDPHMMParams"
,
"src.skill_space.NIWPrior"
,
"collections.OrderedDict"
,
"src.model.VAEConfig"
,
"torch.LongStorage"
,
"torch.device"
,
"torch._utils._rebuild_tensor_v2"
How to fix it?
11 MB
xet
Add vae_with_hmm_round2.pt
3 months ago
vae_with_hmm_round3.pt
pickle
Detected Pickle imports (10)
"torch.float32"
,
"torch.DoubleStorage"
,
"torch.FloatStorage"
,
"src.skill_space.StickyHDPHMMParams"
,
"src.skill_space.NIWPrior"
,
"collections.OrderedDict"
,
"src.model.VAEConfig"
,
"torch.LongStorage"
,
"torch.device"
,
"torch._utils._rebuild_tensor_v2"
How to fix it?
11 MB
xet
Add vae_with_hmm_round3.pt
3 months ago
vae_with_hmm_round4.pt
pickle
Detected Pickle imports (10)
"torch.FloatStorage"
,
"src.skill_space.StickyHDPHMMParams"
,
"torch.LongStorage"
,
"torch._utils._rebuild_tensor_v2"
,
"collections.OrderedDict"
,
"torch.float32"
,
"torch.DoubleStorage"
,
"src.model.VAEConfig"
,
"src.skill_space.NIWPrior"
,
"torch.device"
How to fix it?
11 MB
xet
Add vae_with_hmm_round4.pt
3 months ago
