Downloads · 30 days
3
8% of all-time downloads
ezhoureal/lewm-ensemble
lewm-ensemble is a machine learning model from ezhoureal. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for pytorch.
This repository stores LeWM predictor ensemble checkpoints for PushT auto-compilation uncertainty experiments.
Downloads · 30 days
3
8% of all-time downloads
All-time downloads
40
Public
Repo size
145 MB
Likes
0
Public
Click a slice to open those files.
.pt217 MB · 100%
From the Hugging Face model README
This repository stores LeWM predictor ensemble checkpoints for PushT auto-compilation uncertainty experiments.
lewm_pusht_ensemble_00_seed3072: seed 3072, trained for 20 epochs, final checkpoint members/lewm_pusht_ensemble_00_seed3072/weights_epoch_20.pt.quentinll/lewm-pusht.members/lewm_pusht_ensemble_00_seed3072/weights_epoch_20.pt: final trained PyTorch checkpoint for the first trained ensemble member.members/lewm_pusht_ensemble_00_seed3072/config.json: resolved training/model configuration saved with the checkpoint.members/lewm_pusht_ensemble_00_seed3072/config.yaml: Hydra-style training configuration for the member.ensemble_manifest.json: sanitized manifest for the partial ensemble upload.The next trained ensemble member from the local run was still pending when this upload was prepared.