NickleDave / visual-search-nets

neural network models of visual search behavior
BSD 3-Clause "New" or "Revised" License
6 stars 2 forks source link

add ability to specify whether model should be `pretrained`, and to specify a `weights_path` #87

Closed NickleDave closed 3 years ago

NickleDave commented 3 years ago

to be able to load models trained e.g. on different datasets besides ImageNet, or to perform transfer learning on a randomly initialized model as a control