kogolobo / optimizer_comparison

MIT License
0 stars 1 forks source link

Add pretrained model code #9

Open kogolobo opened 8 months ago

kogolobo commented 8 months ago

We have decided to use pretrained models: ResNet-50 for Image Classification, and DistillBERT for NLP tasks.

We want to modify our codebase to handle those models.