-
- Downloads
First implementation with Lightning
Showing
- bob/learn/pytorch/architectures/ResNet.py 4 additions, 0 deletionsbob/learn/pytorch/architectures/ResNet.py
- bob/learn/pytorch/architectures/iresnet.py 218 additions, 0 deletionsbob/learn/pytorch/architectures/iresnet.py
- bob/learn/pytorch/architectures/lenet.py 40 additions, 0 deletionsbob/learn/pytorch/architectures/lenet.py
- bob/learn/pytorch/datasets/tfrecord/__init__.py 1 addition, 0 deletionsbob/learn/pytorch/datasets/tfrecord/__init__.py
- bob/learn/pytorch/datasets/tfrecord/example_pb2.py 528 additions, 0 deletionsbob/learn/pytorch/datasets/tfrecord/example_pb2.py
- bob/learn/pytorch/datasets/tfrecord/tfrecord.py 165 additions, 0 deletionsbob/learn/pytorch/datasets/tfrecord/tfrecord.py
- bob/learn/pytorch/head/__init__.py 1 addition, 0 deletionsbob/learn/pytorch/head/__init__.py
- bob/learn/pytorch/head/arcface.py 41 additions, 0 deletionsbob/learn/pytorch/head/arcface.py
- bob/learn/pytorch/scripts/lightning.py 82 additions, 0 deletionsbob/learn/pytorch/scripts/lightning.py
- bob/learn/pytorch/test/test_lightning.py 92 additions, 0 deletionsbob/learn/pytorch/test/test_lightning.py
- bob/learn/pytorch/trainers/__init__.py 1 addition, 0 deletionsbob/learn/pytorch/trainers/__init__.py
- bob/learn/pytorch/trainers/lightning.py 105 additions, 0 deletionsbob/learn/pytorch/trainers/lightning.py
- setup.py 37 additions, 42 deletionssetup.py
Loading
Please register or sign in to comment