prototorch_models/prototorch/models
2021-05-19 16:57:51 +02:00
..
__init__.py Custom non-gradient training 2021-05-18 19:49:16 +02:00
abstract.py Add validation and test logic 2021-05-19 16:30:19 +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 Refactor into shared_step 2021-05-19 16:57:51 +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 Close matplotlib figure on train end 2021-05-18 10:13:22 +02:00