 nagadomi
|
c9b0fe7f41
Use w2nn.load_model to use cudnn #369
|
vor 4 Jahren |
|
nagadomi
|
aea254eab5
Rename RandomBinaryCriterion to LBPCriterion
|
vor 6 Jahren |
|
nagadomi
|
b3fb286258
lbp loss: multilayer support
|
vor 6 Jahren |
|
nagadomi
|
4f07b74cc5
Fix resume
|
vor 6 Jahren |
|
nagadomi
|
bdafab9e10
Add -resume_epoch option
|
vor 6 Jahren |
|
nagadomi
|
0d51c40f6b
lbp 512->128
|
vor 6 Jahren |
|
nagadomi
|
89c8f5db8e
Add aux_lbp; Fix oracle
|
vor 6 Jahren |
|
nagadomi
|
f317545732
aux_huber and lbp loss
|
vor 6 Jahren |
|
nagadomi
|
56536ac133
Add support for AuxiliaryLoss
|
vor 6 Jahren |
|
nagadomi
|
93cd40a53c
merge random erasing
|
vor 7 Jahren |
|
nagadomi
|
4d3d123d72
Merge branch 'dev'
|
vor 8 Jahren |
|
nagadomi
|
bb0fc3a1d3
Add -validation_filename_split option
|
vor 8 Jahren |
|
nagadomi
|
063474b2ea
Fix clearState
|
vor 8 Jahren |
|
nagadomi
|
b7e116de54
Add support for multi GPU training (data parallel)
|
vor 8 Jahren |
|
nagadomi
|
202453d6c5
Use same distributions for train/validate in user method
|
vor 8 Jahren |
|
nagadomi
|
cdafbf00ae
Add BCE(binary cross entropy) loss for segmentation
|
vor 8 Jahren |
|
nagadomi
|
dac1b89750
Fix division by zero error in validate()
|
vor 8 Jahren |
|
nagadomi
|
8b5ccbed08
Merge branch 'dev' of into dev
|
vor 8 Jahren |
|
nagadomi
|
763f5ddcab
Fix crop bug in rare case
|
vor 8 Jahren |
|
nagadomi
|
29260ede24
fix batchwise psnr
|
vor 8 Jahren |
|
nagadomi
|
b111901cbb
add -update_criterion option for back compatible
|
vor 8 Jahren |
|
nagadomi
|
0eccbc6555
Add support for MSE loss
|
vor 8 Jahren |
|
nagadomi
|
43a9b58fcb
Add GCN option for user method
|
vor 8 Jahren |
|
nagadomi
|
d2cfb8f104
Add L1 criterion. Change the criterion of updating model
|
vor 8 Jahren |
|
nagadomi
|
bdaca16c67
perfomance tuning
|
vor 8 Jahren |
|
nagadomi
|
b066761cdc
Add data augmentation for user method
|
vor 8 Jahren |
|
nagadomi
|
4998078d2a
use threads.safe to avoid deadlock problem
|
vor 8 Jahren |
|
nagadomi
|
66bff68ef1
split first
|
vor 8 Jahren |
|
nagadomi
|
06e08253e4
Fix some multhread bug
|
vor 8 Jahren |
|
nagadomi
|
a14e6acec3
mutex
|
vor 8 Jahren |
