search

noowad93 / MIMO-pytorch

PyTorch Implementation of MIMO (ICLR 2021)
16 stars 5 forks source link

readme

MIMO-pytorch (WIP)

PyTorch implementation of MIMO proposed in Training independent subnetworks for robust prediction.

Model Training

python train.py

References

@inproceedings{havasi2021training,
  author = {Marton Havasi and Rodolphe Jenatton and Stanislav Fort and Jeremiah Zhe Liu and Jasper Snoek and Balaji Lakshminarayanan and Andrew M. Dai and Dustin Tran},
  title = {Training independent subnetworks for robust prediction},
  booktitle = {International Conference on Learning Representations},
  year = {2021},
}