9 lines
167 B
Python
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)
|