Skip to content

rossning92/ai-lyrics-writing

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

(注:很基本的 RNN 模型,预测效果一般,仅作为学习使用!)

安装关联

  • PyTorch: pip3 install torch==1.8.1+cu102 torchvision==0.9.1+cu102 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html (请根据你的环境选择对应的版本安装 https://pytorch.org/)
  • Tensorboard: pip3 install tensorboard
  • NumPy: pip3 install numpy

训练 & 使用模型预测

在命令行中输入 python3 main.py,根据提示训练或者使用模型做预测。

使用 Tensorboard 查看训练进度

开启一个新的控制台,输入 tensorboard --logdir=runs

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages