文件名称:pytorch-hebbian:PyTorch中用于Hebbian学习的轻量级且灵活的框架
文件大小:316KB
文件格式:ZIP
更新时间:2024-06-12 08:27:37
Python
PyTorch希伯来语 一个基于PyTorch Ignite的Hebbian学习轻量级框架。 在NeurIPS 2020大会上发表。 目录 关于该项目 PyTorch中用于Hebbian学习的轻量级且灵活的框架。 入门 要启动并运行本地副本,请遵循以下简单步骤。 先决条件 根据您的系统(OS / GPU / CUDA支持),您可能需要手动安装特定的PyTorch版本。 请访问PyTorch网站以获取更多信息。 安装 克隆存储库: pip install git+https://github.com/Joxis/pytorch-hebbian.git 用法 类似于经典的PyTorch管道,首先启动模型,数据集和数据加载器。 接下来,框架要求您指定学习规则,优化器和培训器。 然后,您可以开始学习Hebbian。 简单的例子 # Creating the model model =
【文件预览】:
pytorch-hebbian-master
----requirements_gpu.txt(140B)
----requirements.txt(130B)
----requirements_base.txt(108B)
----examples()
--------supervised.py(8KB)
--------data.py(3KB)
--------test.py(2KB)
--------models.py(3KB)
--------hebbian.py(11KB)
--------params()
--------hebbian_simple.py(1KB)
----LICENSE(1KB)
----setup.py(989B)
----README.md(4KB)
----docs()
--------neural_net-local.png(137KB)
--------ml6_logo.png(27KB)
--------visualizers.jpg(143KB)
----.gitignore(1KB)
----pytorch_hebbian()
--------config.py(558B)
--------evaluators.py(5KB)
--------trainers.py(9KB)
--------metrics()
--------utils.py(5KB)
--------__init__.py(0B)
--------nn()
--------handlers()
--------optimizers()
--------learning_rules()