1
0
Fork 0
mirror of synced 2024-06-01 10:39:30 +12:00
Commit graph

589 commits

Author SHA1 Message Date
nagadomi 05bc54fa12 Add support for identity initializer for dilated convolution, and refactor 2017-04-21 08:54:39 +09:00
nagadomi 5fe6cf6ef8 Merge branch 'dev' 2017-04-16 04:40:02 +09:00
nagadomi fe885a3466 Fix -pairwise_y_binary 2017-04-16 04:38:53 +09:00
nagadomi 4d3d123d72 Merge branch 'dev' 2017-04-15 16:30:34 +09:00
nagadomi bb0fc3a1d3 Add -validation_filename_split option 2017-04-15 16:29:38 +09:00
nagadomi e371955cc7 Merge branch 'dev' 2017-04-13 17:38:30 +09:00
nagadomi 063474b2ea Fix clearState 2017-04-13 17:38:18 +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 6ba6cfe1ff Fix gpu option 2017-04-01 03:55:09 +09:00
nagadomi 202453d6c5 Use same distributions for train/validate in user method 2017-02-26 09:03:38 +09:00
nagadomi ed12b43477 Change downsampling filters in pairwise_scale() 2017-02-26 09:01:26 +09:00
nagadomi 34874ec733 Merge branch 'dev' of into dev 2017-02-23 08:51:14 +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 39c7ce1ea7 merge 2017-02-23 08:11:38 +09:00
nagadomi dac1b89750 Fix division by zero error in validate() 2017-02-23 08:10:25 +09:00
nagadomi a121eb39cb Fix division by zero error in validate() 2017-02-23 08:08:18 +09:00
nagadomi d8b7df4505 add support for jaccard in benchmark 2017-02-21 21:39:56 +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 6353f73966 Merge pull request #170 from nagadomi/icc_profile
Add support for icc profile
2017-02-12 17:25:01 +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 b65bed9418 Merge branch 'master' into dev 2017-02-11 01:18:55 +09:00
nagadomi d779a9d47a Merge branch 'dev' 2017-02-11 01:17:49 +09:00
nagadomi c3b01ecc02 Docker: Stop installing cudnn
It is already installed on kaixhin/cuda-torch and latest cudnn does not work with old cunn.
2017-02-07 15:46:26 +09:00
nagadomi 00600ce10d Use cdnjs.cloudflare instead of ajax.googleapis because it's unstable in China. 2017-01-30 06:02:25 +09:00
nagadomi cf2493a1a4 Add support for resnet_14l in tools/export.lua; Add resnet_14l.prototxt 2017-01-26 05:43:55 +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
nagadomi 6051dad793 Fix a bug in ClippedMSECriterion:backward 2017-01-09 12:55:50 +09:00
nagadomi 742c9fdd95 disable gcn 2017-01-09 12:51:45 +09:00
nagadomi 02cf265d48 Reduce memory usage in benchmark 2017-01-04 19:53:17 +09:00
nagadomi 88ddfbc23e Update index.de.html 2017-01-01 04:31:25 +09:00
Benajmin Franz dc08ef693a update german translation 2016-12-31 03:02:21 +01:00
nagadomi 43a9b58fcb Add GCN option for user method 2016-12-25 20:17:47 +09:00
nagadomi a95f41f464 Add pretraind srresnet_12l model; Update benchmark 2016-12-08 06:15:23 +09:00
nagadomi a47e0a9804 Add checking for malformed images 2016-12-05 10:37:42 +09:00
nagadomi d2cfb8f104 Add L1 criterion. Change the criterion of updating model 2016-12-05 10:32:26 +09:00
nagadomi 3c28debad9 Update README 2016-11-09 03:26:04 +09:00
nagadomi 61f2abd632 remove unused code 2016-11-09 03:10:54 +09:00
nagadomi f088af9606 add srresnet_12l 2016-11-09 03:06:57 +09:00
nagadomi c8271af3ab add srresnet_12l 2016-11-09 03:06:26 +09:00