Jensun Ravichandran
db965541fd
Update example
2021-05-25 20:57:54 +02:00
Jensun Ravichandran
d091dea6a1
Update tutorial
2021-05-25 20:54:07 +02:00
Jensun Ravichandran
d411e52be4
Refactor non-gradient-lvq models into lvq.py
2021-05-25 20:37:34 +02:00
Alexander Engelsberger
32d6f95db0
Add RSLVQ and LikelihoodLVQ
2021-05-25 20:26:15 +02:00
Jensun Ravichandran
139109804f
[BUGFIX] Use _forward in LVQ1 and LVQ21
2021-05-25 17:43:37 +02:00
Alexander Engelsberger
2cc11ae2e3
Ignore .pt files
2021-05-25 16:46:31 +02:00
Alexander Engelsberger
72e064338c
Use 'num_' in all variable names
2021-05-25 15:41:10 +02:00
Alexander Engelsberger
e7e6bf9173
Fix failing example
2021-05-21 18:54:47 +02:00
Alexander Engelsberger
2aa631f4e6
Improve example test script (with failing example)
2021-05-21 18:48:37 +02:00
Alexander Engelsberger
c6992da123
Travis didn't use the dev branch
2021-05-21 18:35:00 +02:00
Alexander Engelsberger
dcbd0c1e5c
Travis didn't use the dev branch
2021-05-21 18:30:10 +02:00
Alexander Engelsberger
b8bca71206
[Travis] Use dev branch of prototorch for tests.
2021-05-21 18:22:29 +02:00
Alexander Engelsberger
419eca46af
Add example test script
2021-05-21 18:16:17 +02:00
Alexander Engelsberger
5b12629bd9
All examples use argparse
2021-05-21 17:55:55 +02:00
Alexander Engelsberger
b60db3174a
LightningCLI Example.
2021-05-21 17:13:15 +02:00
Alexander Engelsberger
8ce18f83ce
Add prototype_initializer function to GLVQ
...
This allows overwriting it inside subclasses.
2021-05-21 17:13:10 +02:00
Alexander Engelsberger
7b4f7d84e0
Update Documentation
...
Clean up project
2021-05-21 15:42:45 +02:00
Jensun Ravichandran
a5e086ce0d
Refactor code
2021-05-21 13:33:57 +02:00
Jensun Ravichandran
0611f81aba
Update models namespace
2021-05-21 13:11:59 +02:00
Jensun Ravichandran
a9382dcd9b
Add get_prototype_grid method
2021-05-21 13:11:48 +02:00
Jensun Ravichandran
0933a88a1b
Fix ImageCBC bug
2021-05-21 13:11:36 +02:00
Jensun Ravichandran
88a34a06ef
[WIP] Update CBC implementation to use SiameseGLVQ
2021-05-20 17:36:00 +02:00
Jensun Ravichandran
49f9a12b5f
Update mnist example
2021-05-20 17:35:07 +02:00
Jensun Ravichandran
de63eaf15a
Fix numpy issue in vis.py
2021-05-20 17:33:19 +02:00
Jensun Ravichandran
16dc3cf4eb
Update image visualization
2021-05-20 16:07:16 +02:00
Jensun Ravichandran
df061cc2ff
Refactor code
2021-05-20 14:40:02 +02:00
Alexander Engelsberger
969fb34cc3
Accumulate test loss
2021-05-20 14:20:23 +02:00
Alexander Engelsberger
0204f5eab6
Log test accuracy.
2021-05-20 14:03:31 +02:00
Alexander Engelsberger
b7fc5df386
Log test loss.
2021-05-20 13:47:20 +02:00
Alexander Engelsberger
faf1a88f99
[Bugfix] Remove optimzer_idx from validation and test.
2021-05-20 13:17:27 +02:00
Jensun Ravichandran
5ffbd43a7c
Refactor into shared_step
2021-05-19 16:57:51 +02:00
Jensun Ravichandran
fdf9443a2c
Add validation and test logic
2021-05-19 16:30:19 +02:00
Jensun Ravichandran
7700bb7f8d
[DOC] Ignore Sphinx warnings until prototorch is bumped
...
readthedocs build fails because of missing function (`get_flat`) that's not
available via PyPI yet (See
https://readthedocs.org/projects/prototorch-models/builds/13795474/ ). The
temporary solution until this becomes available is therefore to ignore it and
build the rest of the docs.
2021-05-18 20:08:14 +02:00
Jensun Ravichandran
eefec19c9b
Custom non-gradient training
2021-05-18 19:49:16 +02:00
Jensun Ravichandran
246719b837
[DOC] Add tutorial
2021-05-18 19:41:58 +02:00
Jensun Ravichandran
a14e3aa611
Add argparse to mnist example script
2021-05-18 10:17:51 +02:00
Jensun Ravichandran
00cdacf7ae
Fix example script
2021-05-18 10:15:38 +02:00
Jensun Ravichandran
4957e821f6
Close matplotlib figure on train end
2021-05-18 10:13:22 +02:00
Jensun Ravichandran
538256dcb7
Small changes
2021-05-17 19:37:42 +02:00
Jensun Ravichandran
d812bb0620
Update examples
2021-05-17 17:03:37 +02:00
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
Jensun Ravichandran
7a87636ad7
Update KNN
2021-05-17 16:59:35 +02:00
Jensun Ravichandran
77b7b59bad
Clean visualization callbacks
2021-05-17 16:59:22 +02:00
Jensun Ravichandran
6e7d80be88
[BUGFIX] Fix siamese visualization callback
2021-05-15 12:52:44 +02:00
Jensun Ravichandran
b7684ae512
predict_latent no longer returns numpy
2021-05-15 12:52:16 +02:00
Jensun Ravichandran
ebc42a4aa8
Set gpus=0 in examples
2021-05-15 12:43:00 +02:00
Jensun Ravichandran
c639836537
Update README.md
2021-05-14 22:19:57 +02:00
Jensun Ravichandran
d36d685115
Add links
2021-05-14 12:58:38 +02:00
Alexander Engelsberger
b341096757
Add basic documentation files.
2021-05-13 15:22:32 +02:00
Alexander Engelsberger
0eac2ce326
Examples use GPUs if available.
2021-05-13 15:22:01 +02:00
