Overhaul inheritance - VE - Githubissues

CERC-AAI / Robin

Apache License 2.0

62 stars 8 forks source link

Overhaul inheritance - VE #22

Open daniel-z-kaplan opened 10 months ago

Alexis-BX commented 10 months ago

Right now VEs (in robin/model/multimodal_encoder/) have no parent class

create parent class in VE folder
clean children classes (move as much as possible to parent)
integrate build_vision_tower into class init
remove all calls to children in other files (eg train.py, train()), replace with a type attribute
clean up imports