 nagadomi
|
93cd40a53c
merge random erasing
|
7 vuotta sitten |
|
nagadomi
|
4d3d123d72
Merge branch 'dev'
|
8 vuotta sitten |
|
nagadomi
|
bb0fc3a1d3
Add -validation_filename_split option
|
8 vuotta sitten |
|
nagadomi
|
063474b2ea
Fix clearState
|
8 vuotta sitten |
|
nagadomi
|
b7e116de54
Add support for multi GPU training (data parallel)
|
8 vuotta sitten |
|
nagadomi
|
202453d6c5
Use same distributions for train/validate in user method
|
8 vuotta sitten |
|
nagadomi
|
cdafbf00ae
Add BCE(binary cross entropy) loss for segmentation
|
8 vuotta sitten |
|
nagadomi
|
dac1b89750
Fix division by zero error in validate()
|
8 vuotta sitten |
|
nagadomi
|
8b5ccbed08
Merge branch 'dev' of into dev
|
8 vuotta sitten |
|
nagadomi
|
763f5ddcab
Fix crop bug in rare case
|
8 vuotta sitten |
|
nagadomi
|
29260ede24
fix batchwise psnr
|
8 vuotta sitten |
|
nagadomi
|
b111901cbb
add -update_criterion option for back compatible
|
8 vuotta sitten |
|
nagadomi
|
0eccbc6555
Add support for MSE loss
|
8 vuotta sitten |
|
nagadomi
|
43a9b58fcb
Add GCN option for user method
|
8 vuotta sitten |
|
nagadomi
|
d2cfb8f104
Add L1 criterion. Change the criterion of updating model
|
8 vuotta sitten |
|
nagadomi
|
bdaca16c67
perfomance tuning
|
9 vuotta sitten |
|
nagadomi
|
b066761cdc
Add data augmentation for user method
|
9 vuotta sitten |
|
nagadomi
|
4998078d2a
use threads.safe to avoid deadlock problem
|
9 vuotta sitten |
|
nagadomi
|
66bff68ef1
split first
|
9 vuotta sitten |
|
nagadomi
|
06e08253e4
Fix some multhread bug
|
9 vuotta sitten |
|
nagadomi
|
a14e6acec3
mutex
|
9 vuotta sitten |
|
nagadomi
|
5a3d012f4e
Add random blur method for data augmentation
|
9 vuotta sitten |
|
nagadomi
|
451ee1407f
Stop calculate the instance loss when oracle_rate=0
|
9 vuotta sitten |
|
nagadomi
|
33e6bc888e
Fix the number of threads
|
9 vuotta sitten |
|
nagadomi
|
c2e4bb4380
Support for multi-thread in training
|
9 vuotta sitten |
|
nagadomi
|
7ffd8a51fd
fix message
|
9 vuotta sitten |
|
nagadomi
|
3a70d09921
create new directory when the specified directory is not found
|
9 vuotta sitten |
|
nagadomi
|
81df729a8a
Remove -loss option
|
9 vuotta sitten |
|
nagadomi
|
edac608f18
Add support for user specified pairwise data for universal filter
|
9 vuotta sitten |
|
nagadomi
|
de669d24a4
Change default -validation_crops to 200.
|
9 vuotta sitten |
