Skip to content

Main Training Loop + Models

Sawyer, Haydn (UG - Comp Sci & Elec Eng) requested to merge haydn into main

Adds 5 models and the code to go with training them.

Models:

  • resnet50 (run for 4 epochs)
  • vgg11 (run for 1 epoch before out of memory error, training time was too long to pursue further)
  • densenet121 (run for 4 epochs)
  • densenet121 (run for 20 epochs)
  • inception (run for 4 epochs)

also the vgg11 model is 500mb somehow (I'm sorrrryyy I didn't realize until after I pushed) so enjoy pulling that

Merge request reports