PreTrained works only with PreTrainedFeatureMap

[JeiKeiLim]

As is

• Following configuration will generate an exception error.

backbone:
    # [from, repeat, module, args]
    [
        [-1, 1, UpSample, []],
        [-1, 1, PreTrained, [efficientnet_b0, True]],
        [-1, 1, Conv, [10, 1]
     ]

How it should be

• Above yaml model configuration should just work passing the last feature map of PreTrained to Conv layer.