/pytorch-deep-sets

PyTorch re-implementation of parts of "Deep Sets" (NIPS 2017)

Primary LanguagePythonMIT LicenseMIT

pytorch-deep-sets

PyTorch implementation of parts of "Deep Sets" (NIPS 2017)

Requirements

  • Python 3.6
  • PyTorch 0.3
  • torchvision
  • matplotlib
  • click
  • tqdm