Our source code for EACL2021 workshop: Meme Classification for Tamil Language. We took first place in this task finally!π₯³
Updated: Source code is released!π€©
I will release the code very soon.
βββ MyLoss.py # Impelmentation of some loss function
βββ README.md
βββ __init__.py
βββ args.py # declare some argument
βββ ckpt
βΒ Β βββ README.md
βββ data # store data
βΒ Β βββ README.md
βββ gen_data.py # generate Dataset
βββ install_cli.sh # install required package
βββ logfile # store logfile during training
βββ main.py # train model
βββ model.py # define model
βββ multimodal_attention.py # Implentation of multimodal attention layer
βββ pred_data
βΒ Β βββ README.md
βββ preprocessing.py # preprocess the data
βββ pretrained_weights # store pretrained weights of resnet and xlm-roberta
βΒ Β βββ README.md
βββ run.sh # run model
βββ train.py # define training and validation loop
Use the following command so that you can install all of required packages:
sh install_cli.sh
The first step is to preprocess the data. Just use the following command:
python3 -u preprocessing.py
The second step is to train our model. Use the following command:
nohup sh run.sh > run_log.log 2>&1 &
The final step is inference after training. Use the following command:
nohup python3 -u inference.py > inference.log 2>&1 &
Congralutions! You have got the final results!π€©
If you use our code, please indicate the source.