Skip to content

Comments

Addding T5 model for spatial task#395

Open
premsrit wants to merge 1 commit intoTasksfrom
SPARTQARULE_Generative
Open

Addding T5 model for spatial task#395
premsrit wants to merge 1 commit intoTasksfrom
SPARTQARULE_Generative

Conversation

@premsrit
Copy link
Collaborator

Adding the model for the spatial task. The current implementation is the T5WithLoraGenerativeCLF2 which is the model with the adapter function to use forward of T5 while training and generate while inference in model.py. Example of running command,
python main_rel.py --epoch 12 --train_file SPARTUN --test_file SPARTUN --train_size 20 --test_size 20 --beta 1 --lr 1e-3 --model t5-adapter --check_epoch 1 --version 4

@premsrit premsrit requested a review from DariusNafar March 13, 2024 18:47
@premsrit premsrit self-assigned this Mar 13, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant