12 commits

Author	SHA1	Message	Date
HansBambel	ef14c0e829	diffX and diffY is now an int instead of a Tensor ... Former-commit-id: 342d4e0a8ca19bbee77256caeda0ef6537d746f8	2020-04-24 15:57:53 +02:00
hushenghao	b5a3b207a1	optimize the passed in parameters in the case of bilinear ... Former-commit-id: 6234bd1c4608cac8a64d8658e9af82b60a227859	2020-04-17 23:00:11 +08:00
milesial	490ecf8383	Rework of the transposed conv / bilinear up route ... Former-commit-id: 07debeb9f2621b53ed513e5ab8a0307b4da57767	2020-03-30 20:17:52 -07:00
Anton Kornilov	70e6ac4258	Corrected class Up for model to ONNX exporting ... Former-commit-id: 2dceff8c909b16530e1291118a2165ea95744afd	2019-12-29 18:49:12 +03:00
milesial	f5c2771242	Fix predict script (mistake from last commit) ... Former-commit-id: 9fd386f5929945f3df1868930d392507fee16a55	2019-11-06 13:23:01 +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
milesial	d3150e8923	Fix upsampling issues (maybe) (#38, #32) ... Former-commit-id: 0f45521480d2de8a93407ce43a2a813eefd5d1f6	2018-11-19 11:51:46 +01:00
rht	81234be7da	Move the sigmoid activation to the model itself ... Former-commit-id: e3f8ca7b1ac7c5e9694637a81be260e9b48973b9	2018-11-10 23:11:52 +00:00
milesial	02e2314149	Migration to PyTorch 0.4, code cleanup ... Former-commit-id: c981801ccc3b74047e94c76e67c4ff1f3097226c	2018-06-08 19:27:32 +02:00
milesial	90e988c10f	Fixed bad use of ConvTranspose2D ... Former-commit-id: 245b072a82f2e87a9032a1b87136a9930c008afc	2018-06-07 11:20:35 +03:00
milesial	7ea54febec	Added CLI for predict, cleaned up code, updated README ... Former-commit-id: 77555ccc0925a8fba796ce7e42843d95b6e9dce0	2017-11-30 06:45:19 +01:00