Skip to content

GitLab

Explore

Sign in

Primary navigation

Project

bob.learn.pytorch
- Activity
- Members
- Labels
- Releases
- Model registry
- Environments
- Incidents

Snippets Groups Projects

This is an archived project. Repository and other project resources are read-only.

1f61fbf1

Commit 1f61fbf1 authored 7 years ago by Guillaume HEUSCH

Downloads
- Patches
- Plain Diff

[all] added stuff to train Xiaojiang CNN8 model (architecture, trainer, new Casia dataset)

parent 6515dc97

No related branches found

No related tags found

No related merge requests found

Changes 10

Hide whitespace changes

Inline Side-by-side

Showing

bob/learn/pytorch/architectures/CNN8.py 26 additions, 0 deletions

bob/learn/pytorch/architectures/CNN8.py
bob/learn/pytorch/architectures/__init__.py 2 additions, 0 deletions

bob/learn/pytorch/architectures/__init__.py
bob/learn/pytorch/architectures/utils.py 16 additions, 0 deletions

bob/learn/pytorch/architectures/utils.py
bob/learn/pytorch/datasets/__init__.py 3 additions, 0 deletions

bob/learn/pytorch/datasets/__init__.py
bob/learn/pytorch/datasets/casia_webface.py 54 additions, 5 deletions

bob/learn/pytorch/datasets/casia_webface.py
bob/learn/pytorch/datasets/utils.py 24 additions, 0 deletions

bob/learn/pytorch/datasets/utils.py
bob/learn/pytorch/scripts/train_cnn.py 307 additions, 0 deletions

bob/learn/pytorch/scripts/train_cnn.py
bob/learn/pytorch/trainers/CNNTrainer.py 171 additions, 0 deletions

bob/learn/pytorch/trainers/CNNTrainer.py
bob/learn/pytorch/trainers/__init__.py 1 addition, 0 deletions

bob/learn/pytorch/trainers/__init__.py
setup.py 2 additions, 0 deletions

setup.py

with 606 additions and 5 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