iresnet100.py 302 Bytes
Newer Older
Tiago de Freitas Pereira's avatar
Update  
Tiago de Freitas Pereira committed
1
2
3
4
5
6
7
8
9
10
11
12
13
from bob.bio.face.pytorch.backbones.iresnet import iresnet100

import logging

logger = logging.getLogger(__name__)

logger.info("Loading BACKBONE")
backbone = iresnet100(
    "/idiap/temp/tpereira/bob/data/pytorch/iresnet-91a5de61/iresnet100-73e07ba7.pth"
)

# Reducing the batch size
BATCH_SIZE = 64