Add checkpoint :sparkles:

README.md

@@ -1,3 +1,32 @@
 ---
 license: apache-2.0
+tags:
+- vision
+- classification
+- uncertainty
+datasets:
+- imagenet-1k
 ---
+
+# Packed-Ensembles trained on ImageNet-1k
+
+## How to use
+
+Download [TorchUncertainty](https://torch-uncertainty.github.io/) to use this model.
+
+## License
+
+These weights are provided under the Apache 2.0 license.
+
+## Citation
+
+If you find these weights interesting, please consider citing our [paper](https://arxiv.org/abs/2210.09184):
+
+```text
+@inproceedings{laurent2023packed,
+    title={Packed-Ensembles for Efficient Uncertainty Estimation},
+    author={Laurent, Olivier and Lafage, Adrien and Tartaglione, Enzo and Daniel, Geoffrey and Martinez, Jean-Marc and Bursuc, Andrei and Franchi, Gianni},
+    booktitle={ICLR},
+    year={2023}
+}
+```

config.yaml

@@ -0,0 +1,68 @@
+seed: null
+multi_gpu: true
+logger: true
+checkpoint_callback: null
+enable_checkpointing: true
+default_root_dir: null
+gradient_clip_val: null
+gradient_clip_algorithm: null
+process_position: 0
+num_nodes: 1
+num_processes: null
+devices: null
+gpus: -1
+auto_select_gpus: false
+tpu_cores: null
+ipus: null
+log_gpu_memory: null
+progress_bar_refresh_rate: null
+enable_progress_bar: true
+overfit_batches: 0.0
+track_grad_norm: -1
+check_val_every_n_epoch: 1
+fast_dev_run: false
+accumulate_grad_batches: null
+max_epochs: 105
+min_epochs: null
+max_steps: -1
+min_steps: null
+max_time: null
+limit_train_batches: null
+limit_val_batches: null
+limit_test_batches: null
+limit_predict_batches: null
+val_check_interval: null
+flush_logs_every_n_steps: null
+log_every_n_steps: 50
+accelerator: gpu
+strategy: "A3"
+sync_batchnorm: true
+precision: 16
+enable_model_summary: true
+weights_summary: top
+weights_save_path: null
+num_sanity_val_steps: 2
+resume_from_checkpoint: null
+profiler: null
+benchmark: true
+deterministic: false
+reload_dataloaders_every_n_epochs: 0
+auto_lr_find: false
+replace_sampler_ddp: true
+detect_anomaly: false
+auto_scale_batch_size: false
+prepare_data_per_node: null
+plugins: null
+amp_backend: native
+amp_level: null
+move_metrics_to_cpu: false
+multiple_trainloader_mode: max_size_cycle
+stochastic_weight_avg: false
+terminate_on_nan: null
+batch_size: 1024
+val_split: 0
+num_workers: 12
+num_estimators: 4
+augmentation: 3
+gamma: 1
+resnet: '50'

pe_resnet_50_1k.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5efd244d53e723c77af637f839718debe5e6924fff677ba92629c0f17ef61adb
+size 237086865

performance.md

@@ -0,0 +1,13 @@
+# Performance of this model
+
+## ImageNet-1k
+
+Accuracy: 0.7786
+
+NLL: 1.0376
+
+ECE: 0.1795
+
+Entropy: 2.1925
+
+MI: 0.1322