NickleDave / visual-search-nets

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

ENH: add 'pretrained' and 'weigths_path', fix #87 #88

Closed NickleDave closed 3 years ago

NickleDave commented 3 years ago

adds ability to specify whether model should be pretrained, and to specify a path to pre-trained weights that should be loaded