1
0
Fork 0
mirror of synced 2024-06-29 11:41:29 +12:00
Commit graph

239 commits

Author SHA1 Message Date
nagadomi 6efd7f890e Add cunet for 1x; Remove unused code 2018-10-29 04:07:05 +09:00
nagadomi b28b6172ca clean; Add upcunet_v3 2018-10-28 16:03:52 +09:00
nagadomi 17b8de2d36 Make the performance benchmark practical 2018-10-27 20:38:04 +09:00
nagadomi 4c14b2faf0 Fix clearState in ScaleTable 2018-10-27 20:35:06 +09:00
nagadomi 4c8a47e149 clearState 2018-10-27 09:41:52 +00:00
nagadomi 2d85f70415 Add support for w2nn_valid_input_size 2018-10-27 08:00:23 +00:00
nagadomi 8051535fdc nobias 2018-10-27 06:05:10 +00:00
nagadomi 0b82668359 Add upcunet_v2, fixed the Global AVE problem 2018-10-27 14:59:51 +09:00
nagadomi 1f18d1919a clean 2018-10-25 11:44:55 +00:00
nagadomi 06246e0d78 refactor 2018-10-21 00:56:39 +00:00
nagadomi ef5aa1ccbb SEBlock 2018-10-18 19:49:10 +00:00
nagadomi eea286059f Add ScaleTable for Squeeze and Excitation Networks 2018-10-18 19:35:16 +00:00
nagadomi 4b9196deaa cunet_v5 2018-10-16 13:02:42 +00:00
nagadomi 89c8f5db8e Add aux_lbp; Fix oracle 2018-10-16 13:02:09 +00:00
nagadomi 8065ec9bb2 fix 2018-10-13 20:06:12 +00:00
nagadomi e6da46d08f experimental model 2018-10-14 01:21:23 +09:00
nagadomi f317545732 aux_huber and lbp loss 2018-10-14 01:20:03 +09:00
nagadomi 5365890fa8 Add some modules 2018-10-14 01:18:32 +09:00
nagadomi aef969d64b Add support for criterion arguments in AuxiliaryLossCriterion 2018-10-14 01:15:51 +09:00
nagadomi 20abbbc68e Support for binary model file 2018-10-03 18:42:05 +09:00
nagadomi 56536ac133 Add support for AuxiliaryLoss 2018-10-03 18:41:32 +09:00
nagadomi cf638aea48 Add default option for ClippedMSECriterion 2018-10-03 18:40:04 +09:00
nagadomi 1dc32aaa89 Add w2nn.Print for debug 2018-10-03 18:30:54 +09:00
nagadomi 0fe21eef70 Update random erasing; Support x padding 2018-10-03 18:30:22 +09:00
nagadomi 93cd40a53c merge random erasing 2018-10-03 18:29:05 +09:00
nagadomi fa6ee00624 merge 2018-01-26 08:09:40 +09:00
nagadomi 4ec0b4530d merge 2018-01-26 08:08:52 +09:00
nagadomi cba96e36fb Add -invert_x option for user_method 2017-05-17 08:51:07 +09:00
nagadomi 05bc54fa12 Add support for identity initializer for dilated convolution, and refactor 2017-04-21 08:54:39 +09:00
nagadomi fe885a3466 Fix -pairwise_y_binary 2017-04-16 04:38:53 +09:00
nagadomi bb0fc3a1d3 Add -validation_filename_split option 2017-04-15 16:29:38 +09:00
nagadomi 88e3322296 Add ShakeShakeTable 2017-04-13 17:35:32 +09:00
nagadomi f54dd37848 Merge branch 'dev' 2017-04-10 20:22:37 +09:00
nagadomi b7e116de54 Add support for multi GPU training (data parallel)
train.lua -gpu 1,3,4
When use multi GPU mode, nccl.torch is required.
2017-04-10 20:20:17 +09:00
nagadomi f0fc2c89d1 Add support for grayscale data 2017-04-09 20:53:53 +09:00
nagadomi 71a34393b8 Add support for padding in convert_data.lua 2017-04-08 22:01:13 +09:00
nagadomi eb3926b61c Fix to work on Lua52
web.lua does not work on Lua52 because turbo does not support Lua52 FFI.
2017-03-02 14:47:49 +09:00
nagadomi 780e22cf51 Fix io.read option 2017-03-01 15:17:09 +09:00
nagadomi ed12b43477 Change downsampling filters in pairwise_scale() 2017-02-26 09:01:26 +09:00
nagadomi cdafbf00ae Add BCE(binary cross entropy) loss for segmentation
Sigmoid() output is required.
2017-02-23 08:48:00 +09:00
nagadomi 385020e0e1 Fix a bug of 763f5ddcab 2017-02-19 01:42:40 +09:00
nagadomi 8b5ccbed08 Merge branch 'dev' of into dev 2017-02-12 17:48:44 +09:00
nagadomi c6e3a68974 Optimize flip in user method when -patchs is small 2017-02-12 17:48:21 +09:00
nagadomi 763f5ddcab Fix crop bug in rare case 2017-02-12 17:46:07 +09:00
nagadomi 1db2eeb788 Merge branch 'master' of github.com:nagadomi/waifu2x into dev 2017-02-12 17:27:33 +09:00
nagadomi d6f5e48dab Add support for icc profile 2017-02-12 17:19:03 +09:00
nagadomi 29260ede24 fix batchwise psnr 2017-02-12 02:04:23 +09:00
nagadomi b111901cbb add -update_criterion option for back compatible 2017-02-12 01:56:03 +09:00
nagadomi 3447c6c44c rename srresnet_12l to resnet_14l because it's not the same as SRResNet 2017-01-21 16:53:10 +09:00
nagadomi 0eccbc6555 Add support for MSE loss 2017-01-09 13:00:04 +09:00