Commit graph

17 commits

Author SHA1 Message Date
milesial b83c81f30b Change back to train mode after eval
Former-commit-id: 773ef215d41c1f36dc0ed4159c12df89d792fbc3
2020-05-05 09:15:07 -07:00
whenyd d292e8c6cd Apply sigmoid before calc dice in eval_net()
Former-commit-id: 0da18fda34f29c81968425715e19c5dc76c9ec46
2020-03-14 13:16:52 +08:00
milesial 5f4ce7dba9 Update mask type for muticlass
Former-commit-id: 4dcb7b8440c5f36ff2565c67f56f8f029b589c80
2019-12-13 17:36:12 +01:00
milesial 5a7e934560 Fix eval.py iterating within batch
Former-commit-id: d4b8040593d4bc82774b1dac647c701b2486c477
2019-12-02 12:06:29 +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
dmitrysarov 078b03ffa9 index + 1 in average
Former-commit-id: 1dc82b17f075a74f7ef6b20c888835497bcd690b
2019-01-09 15:01:42 +03:00
rht 81234be7da Move the sigmoid activation to the model itself
Former-commit-id: e3f8ca7b1ac7c5e9694637a81be260e9b48973b9
2018-11-10 23:11:52 +00:00
milesial b1ac7151b4 Switch net to eval mode in eval.py
Former-commit-id: 3ab827241455818c7b81172a426d805f9ace0f8f
2018-09-26 08:57:10 +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 617f334e06 Small changes to README
Former-commit-id: 2c17549a3dc926730e9fcd16ec18610a9e5ec391
2017-11-30 18:50:25 +01:00
milesial 2d427db832 Cleaned code, added image to README.md
Former-commit-id: 3acf1ff8dadb74e95786fb6ddcf1a90de63f5079
2017-11-30 07:47:43 +01:00
milesial 74ee006a7a Added simple predict + submit script 2017-08-21 18:00:07 +02:00
milesial fa40396fff Added simple eval and test CRF 2017-08-19 10:59:51 +02:00