issues
search
shahrukhx01
/
multitask-learning-transformers
A simple recipe for training and inferencing Transformer architecture for Multi-Task Learning on custom datasets. You can find two approaches for achieving this in this repo.
Apache License 2.0
87
stars
11
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
support for text generation task
#5
zhipeng-cai
opened
1 year ago
0
Shared attention
#4
MehwishFatimah
opened
2 years ago
2
Update README.md
#3
shahrukhx01
closed
2 years ago
0
using cpu not gpu
#2
hxtkyne
opened
2 years ago
2
batch[k] = torch.stack([f[k] for f in features])
#1
HaithemH
closed
2 years ago
2