 nagadomi
|
b7e116de54
Add support for multi GPU training (data parallel)
|
vor 8 Jahren |
|
nagadomi
|
d2cfb8f104
Add L1 criterion. Change the criterion of updating model
|
vor 8 Jahren |
|
nagadomi
|
3e77378983
Add SSIMCriterion (supports only forward())
|
vor 8 Jahren |
|
nagadomi
|
cabeeed2a7
Add Clip(0,1) to last layer
|
vor 8 Jahren |
|
nagadomi
|
382d493514
no confidence change for #128
|
vor 9 Jahren |
|
nagadomi
|
a0d8826779
Merge branch 'master' into upconv
|
vor 9 Jahren |
|
nagadomi
|
e50c7d1478
Remove unused modules
|
vor 9 Jahren |
|
nagadomi
|
af74a67bd1
Add -force_cudnn option; support for cuDNN in waifu2x.lua/web.lua
|
vor 9 Jahren |
|
nagadomi
|
8a65db7bab
Change the evaluation metric
|
vor 9 Jahren |
|
nagadomi
|
958dcfaeb8
Add more informative message for CUDA error
|
vor 9 Jahren |
|
nagadomi
|
5bfe504de1
Add error message when the CUDA module loading fails
|
vor 9 Jahren |
|
nagadomi
|
1900ac7500
Use PSNR for evaluation
|
vor 9 Jahren |
|
nagadomi
|
41581a0d55
Remove cleanup_model
|
vor 9 Jahren |
|
nagadomi
|
797b45ae23
Use roundf-like clip for 8 bit-depth image
|
vor 9 Jahren |
|
nagadomi
|
3b2aa67e55
Improve web interface
|
vor 9 Jahren |
|
nagadomi
|
9ddee6088d
Add WeightedHuberCriterion
|
vor 9 Jahren |
|
nagadomi
|
3abc5a03e3
refactor
|
vor 9 Jahren |
