torch_brain

用 PyTorch 快速搭建神经数据模型,省去数据预处理和训练样板代码

开源 unknown 模型训练
访问官网 ↗ GitHub ↗ 文档 ↗
GitHub 星标 ★ 101
维护状态 活跃
是否开源 是
定价模式 unknown

项目预览

项目数据

分类模型训练
开发团队neuro-galaxy
所属国家
定价模式unknown
价格说明定价信息待确认
访问状态
是否开源是
开源协议Apache-2.0
主要语言Python
技术栈/模型
GitHub 星标★ 101
30天Star增速
HF 下载量
上线时间2024-06-24 00:00:00
代码更新—
仓库体积—
本站数据更新—
维护状态活跃
中文支持
访问方式
移动端支持
综合评分
收录时间2026-10-17
浏览次数0

项目介绍

torch_brain 是一个基于 PyTorch 的深度学习库,专门用于神经科学数据的建模与训练。它解决的是神经数据(如电生理记录、钙成像、行为时间序列)格式多样、预处理繁琐、模型搭建重复的问题,让研究者不必从零编写数据加载、时间对齐和训练循环。核心能力包括:面向神经数据的统一数据集接口,支持多通道时间序列的批处理与掩码;内置多种常用网络结构(如 RNN、Transformer、卷积编码器)和训练工具;提供配置化实验管理,方便复现和调参。项目处于早期阶段,代码量不大,适合作为神经数据深度学习的起点或教学示例,也可在此基础上扩展自定义模型。

Star 增长趋势

本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整

正在加载星标数据…

技术标签

使用教程

—

核心亮点

  • 针对神经数据提供统一数据接口,减少时间对齐和批处理的手写代码
  • 内置常用网络模块和训练循环,可快速搭建基线模型
  • 基于 PyTorch 生态,便于复用现有层、优化器和 GPU 加速

不足之处

  • 项目早期,文档和示例较少,上手需要读源码
  • 社区规模小,预训练模型和教程资源有限

适用场景

  • 神经科学实验室分析电生理或钙成像数据,快速训练解码模型
  • 深度学习课程中作为神经数据建模的教学案例
  • 研究人员复现或对比不同网络结构在神经数据上的表现

替代项目

NeuroPy、Braindecode

上一篇:DeepGNN

下一篇:没有了!

同类项目推荐

Long-RL 开源

让强化学习轻松驾驭超长序列,训练更稳更快

Long-RL: Scaling RL to Long Sequences (NeurIPS 2025)

★ 729 2026-08-09
TPA 开源

把注意力复杂度从平方降到线性,长序列不再卡顿

[NeurIPS 2025 Spotlight] TPA: Tensor ProducT ATTenTion Transformer (https://arxiv.or···

★ 464 2026-08-09
minimind 开源

两小时造出你的小模型,练手入门不求人

Train a 64M-parameter LLM from scratch in just 2h!

★ 63828 2026-08-09
pruna 开源

一键优化 AI 模型,推理提速又省资源。

Pruna is a model optimization framework built for developers, enabling you to delive···

★ 1317 2026-08-13