milesial
|
5f4ce7dba9
|
Update mask type for muticlass
Former-commit-id: 4dcb7b8440c5f36ff2565c67f56f8f029b589c80
|
2019-12-13 17:36:12 +01:00 |
|
milesial
|
4e1f0398a1
|
Remove assert for multiclass
Former-commit-id: 16809ec7ba14f67bfa5e5a095d3b512ef669a964
|
2019-12-13 15:58:45 +01:00 |
|
milesial
|
9d7be6e234
|
Cleanup + now using tensorboard
Former-commit-id: 79928c84cdf990ef6fe1043a3e4f74b9cc252642
|
2019-11-23 17:56:14 +01:00 |
|
milesial
|
35f955cbf8
|
Now using utils.data.Dataset
Former-commit-id: c75d9c075e18add5cd8683faf827937393bf2c94
|
2019-11-23 14:22:42 +01:00 |
|
milesial
|
8ed1e09b2a
|
Better support for multiclass
Former-commit-id: 76bebf5f241f579fda7048f5e4a87ee9d49aa423
|
2019-10-30 19:54:57 +01:00 |
|
milesial
|
4c0f0a7a7b
|
Global cleanup, better logging and CLI
Former-commit-id: ff1ac0936c118d129bc8a8014958948d3b3883be
|
2019-10-26 23:17:48 +02:00 |
|
rht
|
81234be7da
|
Move the sigmoid activation to the model itself
Former-commit-id: e3f8ca7b1ac7c5e9694637a81be260e9b48973b9
|
2018-11-10 23:11:52 +00:00 |
|
milesial
|
341824fb06
|
Switch net to train mode in train.py
Former-commit-id: ab255c4b3b823dfe8681083aeac0ee78c1a54780
|
2018-09-26 08:58:49 +02:00 |
|
milesial
|
02e2314149
|
Migration to PyTorch 0.4, code cleanup
Former-commit-id: c981801ccc3b74047e94c76e67c4ff1f3097226c
|
2018-06-08 19:27:32 +02:00 |
|
milesial
|
8008b77af6
|
Removed unused function and general cleanup
Former-commit-id: c34a455f1722e0b899e9e92c7766b83a9a641980
|
2018-04-09 05:15:24 +02:00 |
|
milesial
|
2d427db832
|
Cleaned code, added image to README.md
Former-commit-id: 3acf1ff8dadb74e95786fb6ddcf1a90de63f5079
|
2017-11-30 07:47:43 +01:00 |
|
milesial
|
7ea54febec
|
Added CLI for predict, cleaned up code, updated README
Former-commit-id: 77555ccc0925a8fba796ce7e42843d95b6e9dce0
|
2017-11-30 06:45:19 +01:00 |
|
milesial
|
932e1c3cf5
|
Added simple trained model
Former-commit-id: e87526c043c447e5c8d95ee8314fcb7f97c13ce2
|
2017-08-23 17:38:55 +02:00 |
|
milesial
|
fa40396fff
|
Added simple eval and test CRF
|
2017-08-19 10:59:51 +02:00 |
|
milesial
|
4063565295
|
Created a basic train loop + changed a bit loss and utils
|
2017-08-17 21:16:19 +02:00 |
|
milesial
|
c8c82204bf
|
First prototype
slow, no gpu, no validation
|
2017-08-16 14:24:29 +02:00 |
|