https://github.com/namiyousef/multi-task-learning

Repository for multi task learning
https://github.com/namiyousef/multi-task-learning

mtl pytorch pytorch-implementation torch torchvision

Last synced: about 1 month ago
JSON representation

Repository for multi task learning

• Host: GitHub
• URL: https://github.com/namiyousef/multi-task-learning
• Owner: namiyousef
• Created: 2021-12-17T16:59:51.000Z (about 3 years ago)
• Default Branch: main
• Last Pushed: 2022-05-07T08:40:47.000Z (over 2 years ago)
• Last Synced: 2024-08-04T03:12:03.428Z (5 months ago)
• Topics: mtl, pytorch, pytorch-implementation, torch, torchvision
• Language: Python
• Homepage:
• Size: 578 KB
• Stars: 2
• Watchers: 3
• Forks: 2
• Open Issues: 7
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# Multi-Task Learning

How to run main code:

- Clone the repository.

- Create a new environment from `requirements.txt`.

- Run `main.py`.

How to use this as a library:

- You can import functions from from the different files. The code is designed to behave like a module.

- In particular, the function `get_prebuilt_model` is very useful if you want to load in-built (default) models.

- You can add new defaults based on configurations that you build.

- It is also possible to build models in a bespoke way.

How to run legacy code:

- The legacy code behaves slightly differently to the main code. It is included for reference.

- In particular, `main_colab.py` and `colab_continue_train.py` are useful if you are running on Colab and have trouble with runtimes restarting. You can use them to save models and then re-train from the previously saved state.

- `main.py` from `legacy` should not be used.