
Reformer-PyTorch: 实施改革者
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
Reformer-PyTorch: 实施改革者是一个致力于将Google提出的革命性Transformer模型变种——Reformer用PyTorch框架实现的开源项目,旨在优化大规模序列数据处理效率。
在Pytorch环境中进行重整火炬实施的工作正在进行中。这段代码已在Python 3.7.5和Pytorch 1.4.0上进行了测试,并基于一个框架建立。为了安装必要的依赖,请运行命令 `pip install -r requirements.txt`。
如果需要训练数据集,您可以修改trainer.py或model/model.py文件,但在此之前建议您先熟悉pytorch-lightning库的使用方法。自定义复制任务和音乐数据集已经在datasets/dataloader.py中实现,并可以根据需求进行调整。请将配置yaml文件放置在config目录下;有关基本框架的信息,请参考提供的yaml文件。
运行代码时,请使用如下命令:`python3 trainer.py -c \path\to\config\yaml -n name_of_run -b batch_size -f fast_dev_run -v version`,根据实际情况调整相关参数。
全部评论 (0)
还没有任何评论哟~


