import tensorflow as tf

    y_net = net(x)

    if use_pretrained or pretrained_model_file_path:
        from .tensorflowcv.model_provider import init_variables_from_state_dict
        with tf.Session() as sess:
            from .tensorflowcv.model_provider import load_state_dict
            if pretrained_model_file_path:
                init_variables_from_state_dict(
