transferlearning icon indicating copy to clipboard operation
transferlearning copied to clipboard
verified publisher icon jindongwang

Fix: Add `use_bottleneck` check in `predict`.

Open miaotony opened this issue 1 year ago • 0 comments

Check if use_bottleneck is true before calling self.bottleneck_layer(features) in model.predict function.

miaotony avatar Apr 22 '24 19:04 miaotony