:sparkles: Add model

README.md

@@ -1,3 +1,19 @@
 ---
 license: apache-2.0
+tags:
+- vision
+- classification
+- uncertainty
+datasets:
+- cifar-10
 ---
+
+# BatchEnsemble ResNet trained on CIFAR-10
+
+## How to use
+
+Download [TorchUncertainty](https://torch-uncertainty.github.io/) - [GitHub](https://github.com/ENSTA-U2IS/torch-uncertainty) to use this model.
+
+## License
+
+These weights are provided under the Apache 2.0 license.

bch_resnet18_c10.ckpt

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

config.yaml

@@ -0,0 +1,91 @@
+accelerator: null
+accumulate_grad_batches: null
+amp_backend: null
+amp_level: null
+arch: 18
+auto_augment: null
+auto_lr_find: false
+auto_scale_batch_size: false
+auto_select_gpus: null
+batch_size: 128
+benchmark: null
+channels_last: false
+check_val_every_n_epoch: 1
+ckpt: null
+corruption_severity: null
+cutmix_alpha: 0
+cutout: 0
+default_root_dir: null
+detect_anomaly: false
+devices: null
+dist_sim: emb
+dropout_rate: 0.0
+enable_checkpointing: true
+enable_model_summary: true
+enable_progress_bar: true
+enable_resume: false
+evaluate_ood: false
+exp_dir: logs/
+exp_name: batched-resnet18-cifar10
+fast_dev_run: false
+gpus: 1
+gradient_clip_algorithm: null
+gradient_clip_val: null
+groups: 1
+in_channels: 3
+inference_mode: true
+ipus: null
+kernel_tau_max: 1.0
+kernel_tau_std: 0.5
+limit_predict_batches: null
+limit_test_batches: null
+limit_train_batches: null
+limit_val_batches: null
+log_every_n_steps: 50
+log_graph: false
+log_plots: false
+logger: true
+max_epochs: 75
+max_steps: -1
+max_time: null
+min_epochs: null
+min_steps: null
+mixmode: elem
+mixtype: erm
+mixup_alpha: 0
+move_metrics_to_cpu: false
+multiple_trainloader_mode: max_size_cycle
+n_splits: 10
+num_classes: 10
+num_estimators: 4
+num_nodes: 1
+num_processes: null
+num_sanity_val_steps: 2
+num_workers: 4
+opt_temp_scaling: false
+overfit_batches: 0.0
+plugins: null
+precision: bf16
+profiler: null
+reload_dataloaders_every_n_epochs: 0
+replace_sampler_ddp: true
+resume_from_checkpoint: null
+seed: null
+strategy: null
+style: cifar
+summary: false
+sync_batchnorm: false
+test: null
+test_alt: null
+tpu_cores: null
+track_grad_norm: -1
+train_over: 4
+use_cv: false
+use_entropy: false
+use_logits: false
+use_mi: false
+use_variation_ratio: false
+val_check_interval: null
+val_split: 0.1
+val_temp_scaling: false
+version: batched