Alexander Engelsberger
|
76fea3f881
|
chore: update all examples to pytorch 1.6
|
2022-05-17 12:03:43 +02:00 |
|
Alexander Engelsberger
|
d7ea89d47e
|
feat: add simple test step
|
2021-09-10 19:19:51 +02:00 |
|
Alexander Engelsberger
|
7b9b767113
|
fix: training loss is a zero dimensional tensor
Should fix the problem with EarlyStopping callback.
|
2021-06-25 17:07:06 +02:00 |
|
Alexander Engelsberger
|
8956ee75ad
|
test(githooks): Add githooks for automatic commit checks
|
2021-06-16 16:16:34 +02:00 |
|
Jensun Ravichandran
|
5d2a8226ce
|
Update example scripts
|
2021-06-04 22:21:28 +02:00 |
|
Jensun Ravichandran
|
e9d2075fed
|
Sort imports in example scripts
|
2021-05-31 00:52:16 +02:00 |
|
Alexander Engelsberger
|
5b12629bd9
|
All examples use argparse
|
2021-05-21 17:55:55 +02:00 |
|
Jensun Ravichandran
|
d812bb0620
|
Update examples
|
2021-05-17 17:03:37 +02:00 |
|
Jensun Ravichandran
|
ebc42a4aa8
|
Set gpus=0 in examples
|
2021-05-15 12:43:00 +02:00 |
|
Alexander Engelsberger
|
0eac2ce326
|
Examples use GPUs if available.
|
2021-05-13 15:22:01 +02:00 |
|
Jensun Ravichandran
|
59b8ab6643
|
Add knn
|
2021-05-11 17:22:02 +02:00 |
|