liaoxingyu
866a196d19
add more datasets support
2020-09-01 16:13:12 +08:00
liaoxingyu
2c0320da1a
fix bug in pid_dict
...
Summary: id index will be different in different machine.
This will lead to different fc layer when training on different machine.
2020-08-20 15:41:14 +08:00
liaoxingyu
b7a2b1c21a
support DDP sampler and augment
2020-07-06 16:55:23 +08:00
liaoxingyu
e2761f7581
fix style in comm dataset
2020-05-30 16:56:08 +08:00
liaoxingyu
84c733fa85
fix: remove prefetcher, put normalizer in model
...
1. remove messy data prefetcher which will cause confusion
2. put normliazer in model to accelerate training via GPU computing
2020-05-25 23:39:11 +08:00
liaoxingyu
6d96529d4c
fix(data): fix resume training bug
...
fix dataset pid dictionary loading bug when resume training,
data prefetcher will pre-load a batch of data, this will lead to
misalignment of old pid dict and updated pid dict.
We can address this problem by redefine a prefetcher in resume_or_load
2020-05-05 23:20:42 +08:00
liaoxingyu
6a8961ce48
1. upload circle loss and arcface
...
2. finish freeze training
3. update augmix data augmentation
2020-04-05 23:54:26 +08:00
L1aoXingyu
b020c7f0ae
Fix data prefetcher minor bug
2020-02-27 12:16:57 +08:00
L1aoXingyu
12957f66aa
Change architecture:
...
1. delete redundant preprocess
2. add data prefetcher to accelerate data loading
3. fix minor bug of triplet sampler when only one image for one id
2020-02-18 21:01:23 +08:00
L1aoXingyu
327d74ffbb
Update strong baseline result
...
Change data sampler
2020-02-13 00:19:15 +08:00
L1aoXingyu
db6ed12b14
Update sampler code
2020-02-10 07:38:56 +08:00