search

jindongwang / transferlearning

Transfer learning / domain adaptation / domain generalization / multi-task learning etc. Papers, codes, datasets, applications, tutorials.-迁移学习
http://transferlearning.xyz/
MIT License
13.36k stars 3.81k forks source link

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

Open miaotony opened 5 months ago

miaotony commented 5 months ago

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