一个简单且完整的PyTorch框架
PyTorch framework
该框架适合懂一点PyTorch但没有完整加载自己数据跑过实验的人，以及完整跑过实验但代码写的不优雅的人。可以在该框架上直接迁移到自己任务上，只需要更改dataloader和model部分 (还有损失函数)。
代码：https://github.com/CC-21H/PyTorch-framework
一个简单且完整的PyTorch的框架，提供了各种数据的加载以及简单任务的解决方案，易于扩展和迁移。
1.该框架提供了各种数据类型的加载(.wav .mat .jpg .csv .npy)方案。
2.该框架提供了简单分类任务和回归任务的解决方案，以及几个基础模型：CNN、RNN、Attention （ResNet、LSTM、Transformer-encoder)
3.该框架是一个简单且完整的框架，只保留了必要的部分并有详细的注释，方便阅读和理解。
并且解耦了各个模块，易于扩展和迁移。迁移到其他任务上只需要更改dataloader和model部分 (还有损失函数)。
本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！