有关代码的信息和说明
要运行代码,需要Python 3.6和PyTorch 0.2
培训代码位于文件mtl_learning.py而测试代码位于文件mtl_testing.py
mtl_learning.py的代码mtl_learning.py是自我记录的
为了进行训练,数据集应该在pickles/目录中以pickle文件的形式出现。 嵌入文件应位于data/目录中。
如果重新加载模型并继续训练,则应将模型放置在reloads/目录中。
当所有必要的文件和数据都存在时,只需运行python mtl_learning.py进行培训,然后运行python mtl_testing.py进行测试。
培训的输出将显示在名为test_output.txt的文件中,而测试的outputs.txt将显示在test_output.txt
学分
该项目的灵感来自Pasunuru等人(2017)的工作
2022-12-22 21:29:59
10.89MB
Python
1