prototorch_models/prototorch/models
Jensun Ravichandran 81346785bd Cleanup models
Siamese architectures no longer accept a `backbone_module`. They have to be
initialized with an pre-initialized backbone object instead. This is so that the
visualization callbacks could use the very same object for visualization
purposes. Also, there's no longer a dependent copy of the backbone. It is
managed simply with `requires_grad` instead.
2021-05-17 17:00:23 +02:00
..
__init__.py Stop passing component initializers as hparams 2021-05-12 16:36:22 +02:00
abstract.py Cleanup models 2021-05-17 17:00:23 +02:00
callbacks.py Examples use GPUs if available. 2021-05-13 15:22:01 +02:00
cbc.py Examples use GPUs if available. 2021-05-13 15:22:01 +02:00
glvq.py Cleanup models 2021-05-17 17:00:23 +02:00
knn.py Update KNN 2021-05-17 16:59:35 +02:00
neural_gas.py Change optimizer using kwargs 2021-05-11 16:13:00 +02:00
vis.py Clean visualization callbacks 2021-05-17 16:59:22 +02:00