Skip to content

GitLab

Explore

Sign in
Register

Primary navigation

Project

C

COM3015 Group Project
- Activity
- Members
- Labels
- Issues
- Issue boards
- Milestones
- Wiki
- Requirements
- Environments
- Terraform modules
- Incidents

Snippets Groups Projects

fa2c010d

Commit fa2c010d authored 2 years ago by Sawyer, Haydn (UG - Comp Sci & Elec Eng)

Downloads
- Patches
- Plain Diff

Added main training loop, and added the trained models to the model folder

parent 93aa1b14

No related branches found

No related tags found

Loading

Changes 13

Hide whitespace changes

Inline Side-by-side

Showing

.gitignore 1 addition, 0 deletions

.gitignore
Models/densenet121_run20.pt 0 additions, 0 deletions

Models/densenet121_run20.pt
Models/densenet121_run4.pt 0 additions, 0 deletions

Models/densenet121_run4.pt
Models/inception_run4.pt 0 additions, 0 deletions

Models/inception_run4.pt
Models/resnet50_run4.pt 0 additions, 0 deletions

Models/resnet50_run4.pt
Models/vgg11_run1.pt 0 additions, 0 deletions

Models/vgg11_run1.pt
__pycache__/constants.cpython-310.pyc 0 additions, 0 deletions

__pycache__/constants.cpython-310.pyc
__pycache__/datahandler.cpython-310.pyc 0 additions, 0 deletions

__pycache__/datahandler.cpython-310.pyc
constants.py 11 additions, 4 deletions

constants.py
datahandler.py 0 additions, 1 deletion

datahandler.py
main.py 20 additions, 4 deletions

main.py
model.py 34 additions, 2 deletions

model.py
train.py 103 additions, 3 deletions

train.py

with 169 additions and 14 deletions

Loading

0% Loading or .

You are about to add 0 people to the discussion. Proceed with caution.

Finish editing this message first!

Please register or sign in to comment