nagadomi
|
88e3322296
|
Add ShakeShakeTable
|
2017-04-13 17:35:32 +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
|
d2cfb8f104
|
Add L1 criterion. Change the criterion of updating model
|
2016-12-05 10:32:26 +09:00 |
|
nagadomi
|
3e77378983
|
Add SSIMCriterion (supports only forward())
|
2016-10-02 22:22:50 +09:00 |
|
nagadomi
|
cabeeed2a7
|
Add Clip(0,1) to last layer
|
2016-09-14 21:23:29 +09:00 |
|
nagadomi
|
382d493514
|
no confidence change for #128
|
2016-07-27 22:55:56 +09:00 |
|
nagadomi
|
a0d8826779
|
Merge branch 'master' into upconv
|
2016-07-09 09:58:48 +09:00 |
|
nagadomi
|
e50c7d1478
|
Remove unused modules
|
2016-06-13 12:48:11 +09:00 |
|
nagadomi
|
af74a67bd1
|
Add -force_cudnn option; support for cuDNN in waifu2x.lua/web.lua
|
2016-06-12 16:33:50 +09:00 |
|
nagadomi
|
8a65db7bab
|
Change the evaluation metric
|
2016-05-27 16:57:14 +09:00 |
|
nagadomi
|
958dcfaeb8
|
Add more informative message for CUDA error
|
2016-05-06 21:48:53 +09:00 |
|
nagadomi
|
5bfe504de1
|
Add error message when the CUDA module loading fails
|
2016-05-06 02:20:54 +09:00 |
|
nagadomi
|
1900ac7500
|
Use PSNR for evaluation
|
2016-03-12 06:53:42 +09:00 |
|
nagadomi
|
41581a0d55
|
Remove cleanup_model
|
2016-03-12 05:43:04 +09:00 |
|
nagadomi
|
797b45ae23
|
Use roundf-like clip for 8 bit-depth image
Maybe PSNR +0.03 improved by this commit
|
2015-11-08 05:44:14 +09:00 |
|
nagadomi
|
3b2aa67e55
|
Improve web interface
|
2015-11-07 22:54:29 +09:00 |
|
nagadomi
|
9ddee6088d
|
Add WeightedHuberCriterion
|
2015-10-31 04:36:20 +09:00 |
|
nagadomi
|
3abc5a03e3
|
refactor
|
2015-10-28 16:01:07 +09:00 |
|