├── .gitignore ├── LICENSE ├── README.md ├── docs ├── ml6_logo.png ├── neural_net-local.png └── visualizers.jpg ├── examples ├── data.py ├── hebbian.py ├── hebbian_simple.py ├── models.py ├── params │ ├── cifar-10.json │ ├── mnist-fashion-conv.json │ ├── mnist-fashion.json │ └── mnist.json ├── supervised.py └── test.py ├── pytorch_hebbian ├── __init__.py ├── config.py ├── evaluators.py ├── handlers │ ├── __init__.py │ ├── tensorboard_logger.py │ └── tqdm_logger.py ├── learning_rules │ ├── __init__.py │ ├── hebb.py │ ├── krotov.py │ ├── learning_rule.py │ └── oja.py ├── metrics │ ├── __init__.py │ └── unit_convergence.py ├── nn │ ├── __init__.py │ ├── layers.py │ └── loss.py ├── optimizers │ ├── __init__.py │ └── local.py ├── trainers.py └── utils.py ├── requirements.txt ├── requirements_base.txt ├── requirements_gpu.txt └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/README.md -------------------------------------------------------------------------------- /docs/ml6_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/docs/ml6_logo.png -------------------------------------------------------------------------------- /docs/neural_net-local.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/docs/neural_net-local.png -------------------------------------------------------------------------------- /docs/visualizers.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/docs/visualizers.jpg -------------------------------------------------------------------------------- /examples/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/examples/data.py -------------------------------------------------------------------------------- /examples/hebbian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/examples/hebbian.py -------------------------------------------------------------------------------- /examples/hebbian_simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/examples/hebbian_simple.py -------------------------------------------------------------------------------- /examples/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/examples/models.py -------------------------------------------------------------------------------- /examples/params/cifar-10.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/examples/params/cifar-10.json -------------------------------------------------------------------------------- /examples/params/mnist-fashion-conv.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/examples/params/mnist-fashion-conv.json -------------------------------------------------------------------------------- /examples/params/mnist-fashion.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/examples/params/mnist-fashion.json -------------------------------------------------------------------------------- /examples/params/mnist.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/examples/params/mnist.json -------------------------------------------------------------------------------- /examples/supervised.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/examples/supervised.py -------------------------------------------------------------------------------- /examples/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/examples/test.py -------------------------------------------------------------------------------- /pytorch_hebbian/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pytorch_hebbian/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/pytorch_hebbian/config.py -------------------------------------------------------------------------------- /pytorch_hebbian/evaluators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/pytorch_hebbian/evaluators.py -------------------------------------------------------------------------------- /pytorch_hebbian/handlers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pytorch_hebbian/handlers/tensorboard_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/pytorch_hebbian/handlers/tensorboard_logger.py -------------------------------------------------------------------------------- /pytorch_hebbian/handlers/tqdm_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/pytorch_hebbian/handlers/tqdm_logger.py -------------------------------------------------------------------------------- /pytorch_hebbian/learning_rules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/pytorch_hebbian/learning_rules/__init__.py -------------------------------------------------------------------------------- /pytorch_hebbian/learning_rules/hebb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/pytorch_hebbian/learning_rules/hebb.py -------------------------------------------------------------------------------- /pytorch_hebbian/learning_rules/krotov.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/pytorch_hebbian/learning_rules/krotov.py -------------------------------------------------------------------------------- /pytorch_hebbian/learning_rules/learning_rule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/pytorch_hebbian/learning_rules/learning_rule.py -------------------------------------------------------------------------------- /pytorch_hebbian/learning_rules/oja.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/pytorch_hebbian/learning_rules/oja.py -------------------------------------------------------------------------------- /pytorch_hebbian/metrics/__init__.py: -------------------------------------------------------------------------------- 1 | from .unit_convergence import * 2 | -------------------------------------------------------------------------------- /pytorch_hebbian/metrics/unit_convergence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/pytorch_hebbian/metrics/unit_convergence.py -------------------------------------------------------------------------------- /pytorch_hebbian/nn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/pytorch_hebbian/nn/__init__.py -------------------------------------------------------------------------------- /pytorch_hebbian/nn/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/pytorch_hebbian/nn/layers.py -------------------------------------------------------------------------------- /pytorch_hebbian/nn/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/pytorch_hebbian/nn/loss.py -------------------------------------------------------------------------------- /pytorch_hebbian/optimizers/__init__.py: -------------------------------------------------------------------------------- 1 | from .local import * 2 | -------------------------------------------------------------------------------- /pytorch_hebbian/optimizers/local.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/pytorch_hebbian/optimizers/local.py -------------------------------------------------------------------------------- /pytorch_hebbian/trainers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/pytorch_hebbian/trainers.py -------------------------------------------------------------------------------- /pytorch_hebbian/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/pytorch_hebbian/utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/requirements.txt -------------------------------------------------------------------------------- /requirements_base.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/requirements_base.txt -------------------------------------------------------------------------------- /requirements_gpu.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/requirements_gpu.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/julestalloen/pytorch-hebbian/HEAD/setup.py --------------------------------------------------------------------------------