BenchFormers v0.2.0

Features

e7ea6a8 feat: Made it possible to choose a model to train
fafab10 feat: Set pytorch env variable to allocate less gpu memory

Bug Fixes

d4b8aa8 fix: Used --models argument
a6f7563 fix: Return the model as a list to make it iterable
a6e72ac fix: Removed the unused loop
8f5d07d fix: Typo
c9c3036 fix: Correct usage of main.py
26eff72 fix: Brought back Roformer Model
7ee941f fix: Added gradient checkpointing to save GPU memory
e942843 fix: syntax error on logger
63febf1 fix: Removed the problematic line of code
3c969f3 fix: Unterminated string

Documentation

Refactors

c36f594 refactor: Models will be trained sequentially to save gpu memory usage
8e35ee7 refactor: Optimised trainer to use less GPU memory
e8231c9 refactor: Stratify the labels
a05e669 refactor: Formatted code as per PEP requirements
71ce5c2 refactor: Removed unnecessary parameters

Chores

0e654d6 chore: Added targets for evaluation
01e5f0e chore: update logger to default to info
310cd4f chore: Added target for 1k data
a3f9dad chore: Added targets for running the code

Other Changes

cbc3701 git: Workflow that drafts a release when tag is pushed