AIlib2/p2pnetUtils/__init__.py

9 lines
167 B
Python

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