5d21dd2
1
2
3
4
5
6
7
8
9
10
11
import logging logger = logging.getLogger('base') def create_model(opt): model = opt['model'] frame_num = opt['gop'] from .IBSN import Model_VSN as M m = M(opt) logger.info('Model [{:s}] is created.'.format(m.__class__.__name__)) return m
