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
Jensun Ravichandran
8f9c29bd2b
[BUGFIX] Remove incorrect import statement
2021-05-12 16:45:01 +02:00
Jensun Ravichandran
ca39aa00d5
Stop passing component initializers as hparams
...
Pass the component initializer as an hparam slows down the script very much. The
API has now been changed to pass it as a kwarg to the models instead.
The example scripts have also been updated to reflect the new changes.
Also, ImageGMLVQ and an example script `gmlvq_mnist.py` that uses it have also
been added.
2021-05-12 16:36:22 +02:00
Alexander Engelsberger
1498c4bde5
Bump version: 0.1.6 → 0.1.7
2021-05-11 17:18:29 +02:00
Jensun Ravichandran
59b8ab6643
Add knn
2021-05-11 17:22:02 +02:00
Jensun Ravichandran
2a4f184163
Update example scripts
2021-05-11 16:15:08 +02:00
Jensun Ravichandran
265e74dd31
Require prototorch>=0.4.2
2021-05-11 16:14:47 +02:00
Jensun Ravichandran
daad018a78
Update readme
2021-05-11 16:14:23 +02:00
Jensun Ravichandran
eab1ec72c2
Change optimizer using kwargs
2021-05-11 16:13:00 +02:00
Jensun Ravichandran
b38acd58a8
[BUGFIX] Fix visualization callbacks bug
2021-05-11 16:09:27 +02:00
Alexander Engelsberger
e87563e10d
Bump version: 0.1.5 → 0.1.6
2021-05-11 13:41:26 +02:00
Alexander Engelsberger
767206f905
Define minimum prototorch version in setup
2021-05-11 13:41:09 +02:00
Alexander Engelsberger
3fa6378c4d
Add LVQ1 and LVQ2.1 Models.
2021-05-11 13:26:13 +02:00
Alexander Engelsberger
30ee287ecc
Bump version: 0.1.4 → 0.1.5
2021-05-10 17:13:00 +02:00
Alexander Engelsberger
e323f9d4ca
Fix long description for pypi.
2021-05-10 17:12:54 +02:00
Alexander Engelsberger
f49db0bf2c
Bump version: 0.1.3 → 0.1.4
2021-05-10 17:06:28 +02:00
Alexander Engelsberger
db38667306
Fix Travis configuration
2021-05-10 17:06:23 +02:00
Alexander Engelsberger
54a8494d86
Bump version: 0.1.2 → 0.1.3
2021-05-10 17:04:20 +02:00
Alexander Engelsberger
bf310be97c
Bump version: 0.1.1 → 0.1.2
2021-05-10 16:47:33 +02:00
Alexander Engelsberger
32ae1b7862
Add Build Badge.
2021-05-10 16:47:28 +02:00
Jensun Ravichandran
dfddb92aba
Dummy change
2021-05-10 16:47:37 +02:00
Alexander Engelsberger
4a38bb2bfe
Corrected Badge Image Url.
2021-05-10 16:40:08 +02:00
Alexander Engelsberger
6680d4b9df
Add PyPi Badge.
2021-05-10 16:39:02 +02:00
Alexander Engelsberger
1ae2b41edd
Bump version: 0.1.0 → 0.1.1
2021-05-10 16:26:21 +02:00
Alexander Engelsberger
9300a6d14d
Dummy Test to enable CI.
2021-05-10 16:19:51 +02:00
Jensun Ravichandran
3d42876df1
Merge pull request #7 from si-cim/main
...
Merge branch 'dev' into main
2021-05-10 16:07:08 +02:00
Alexander Engelsberger
fbadacdbca
Merge branch 'dev' into main
2021-05-10 16:02:07 +02:00
Alexander Engelsberger
f6e3a37e2b
Bump version: 0.0.0 → 0.1.0
2021-05-10 16:01:46 +02:00
Jensun Ravichandran
34fb8d796f
Setup travis
2021-05-10 16:03:23 +02:00
Jensun Ravichandran
95c21dd9ca
Merge pull request #6 from si-cim/main
...
Pull changes to dev from main
2021-05-10 15:47:55 +02:00