yaringal / multi-task-learning-example

A multi-task learning example for the paper https://arxiv.org/abs/1705.07115
MIT License
838 stars 205 forks source link

multi-task-learning-example pytorch notebook #5

Closed RagMeh11 closed 4 years ago

RagMeh11 commented 5 years ago

Hi! This is a pytorch version of the notebook

Results are marginally lower than the keras version of the notebook.

standard deviations: Keras: [8.6481799024418997, 0.92546439885191034] standard deviations: pytorch: [8.587499618530273, 0.9198104739189148]

RagMeh11 commented 4 years ago

@yaringal Any possibility of merging this to the master branch?