Skip to content
/ mmf Public
forked from facebookresearch/mmf

This fork contains code for CS7643 Deep Learning Final Project. The task is to improve on the SOTA baselines for the Hateful Memes Dataset

License

Notifications You must be signed in to change notification settings

reoneo97/mmf

 
 

Repository files navigation

Cs7643 project

Run

Save Dataset

mmf_convert_hm --zip_file=data.zip --password=1 --bypass_checksum=1

`mmf_run config=projects/hateful_memes/configs/

Run ViTEncBert

mmf_run config=projects/hateful_memes/configs/vilbert/vitencbert.yaml \
    model=vitencbert \
    dataset=hateful_memes \
    run_type=train_val

Code Changes

  • Slight bugfix for wandb logger
  • Added code for the hateful memes dataset
  • Adjust preprocessing code

VITBERT

mmf_run config=projects/hateful_memes/configs/vilbert/vitbert.yaml \
    model=vitbert \
    dataset=hateful_memes \
    run_type=train_val 
  • Copy over embeddings
  • Alignment of attention layers
    • There are 12 layers each for text and image What was done
mmf_run config=projects/hateful_memes/configs/vilbert/vitbert.yaml \
    model=vitbert \
    dataset=hateful_memes \
    run_type=train_val 

Prediction

mmf_predict config=projects/hateful_memes/configs/vilbert/vitencbert.yaml
model=vitencbert
dataset=hateful_memes
run_type=test
checkpoint.resume_file=

About

This fork contains code for CS7643 Deep Learning Final Project. The task is to improve on the SOTA baselines for the Hateful Memes Dataset

Resources

License

Code of conduct

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.0%
  • Other 1.0%