models/__init__.py

from .p2pnet import build

# build the P2PNet model
# set training to 'True' during training
def build_model(args, training=False):
    return build(args, training)