Commit Graph

6 Commits

Author SHA1 Message Date
Piv
58b8e53986 Fix packnet residual block and layers, refactor to support different amount of residual layers
I noticed the number of parameters didn't match up to the paper (~128 million)
Fixed this by doing the following:
 - Kernel size of 1 for 3rd conv2d in residual block
 - Use add rather than concat in residual block
 - Fixed add/concat features in decode layers
 - Fixed final layers -> this also allows features_3d == 16 to work
2021-08-07 21:00:17 +09:30
Piv
3254eef4bf Add compiling packnet model, refactor modules to not duplicate loaders and trainers 2021-07-23 22:41:46 +09:30
Piv
66cbc7faf6 Add pack layer tests, fix unpack_3d layer 2021-07-19 20:37:54 +09:30
Piv
38e7ad069e Refactor load/util, start fixing packnet to support NHWC format 2021-07-19 12:32:56 +09:30
Piv
d8bf493999 More packnet implementation 2021-07-18 19:54:28 +09:30
Piv
603de2bc9f Start adding packnet model 2021-07-18 18:59:25 +09:30