RelationNetworks CLEVR Versions Save

A pytorch implementation for "A simple neural network module for relational reasoning", working on the CLEVR dataset

v1.0.0

5 years ago

Improved functionalities

  • Accuracies comparable with those claimed in the original paper (see below)
  • Released pretrained models for both state-description and from-pixels
  • Added features extraction routines

Accuracies

  • state-description: 97.9%
  • from-pixels: 93.6%

v1.0.0-alpha

6 years ago

Basic functionalities

  • Support for CLEVR dataset with batch loading from disk
  • Both state-description and from-pixel versions supported
  • Exponential increase policy for learning rate and/or batch-size
  • High customizable hyperparameters using json coded configuration file

Accuracies

  • state-description: 98%
  • from-pixels: not trained yet