liaoxingyu
10cbaab155
support finetuning from trained models
...
Summary: add a flag for supporting finetuning model from the trained weights, and it's very useful when performing across domain reid
2020-09-28 17:10:10 +08:00
liaoxingyu
fae128a4db
fix bug in evaluator ( #284 )
...
Summary: Change `Trainer` to `DefaultTrainer` in `tools/train.py` for adapting to new version of `build_evaluator`.
2020-09-25 15:05:56 +08:00
liaoxingyu
07872d4cdb
fix caffe export bug
...
Summary: put tensor to gpu device
2020-08-20 16:28:52 +08:00
liaoxingyu
65169b40bd
support ddp testing
2020-07-30 20:15:28 +08:00
liaoxingyu
64861ea6c8
update config file
2020-07-29 19:48:49 +08:00
liaoxingyu
a8c7e07df6
update deploy docs
2020-07-29 17:47:35 +08:00
liaoxingyu
16655448c2
onnx/trt support
...
Summary: change model pretrain mode and support onnx/TensorRT export
2020-07-29 17:43:39 +08:00
liaoxingyu
fec7abc461
finish v0.2 ddp training
2020-07-06 16:57:43 +08:00
liaoxingyu
8879db3fba
update training instruction
...
Summary: update dataset configuration and training instruction
2020-06-16 19:43:36 +08:00
liaoxingyu
27c48c8f02
update pytorchtocaffe tool
...
Summary: update pytorch to caffe toolchain to support model layers and bash scripts
2020-06-09 14:45:27 +08:00
Christian Clauss
d7cf32a652
Fix a Python syntax error discovered by #77
2020-06-03 14:32:46 +02:00
liaoxingyu
69e4ba901e
update acknowledgement in deploy
2020-06-02 18:12:57 +08:00
liaoxingyu
2f6d999469
add pytorch to caffe converting
...
Summary: update deployment README and support pytorch to caffe converting for basemodel
2020-06-02 18:11:02 +08:00
liaoxingyu
4be4cacb73
fix: add a simple way to reset data prefetcher when resume training
...
use data prefetcher build-in reset function to reload it rather than
redefining a new data prefetcher, otherwise it will introduce other
problems in eval-only mode.
2020-05-09 11:58:27 +08:00
liaoxingyu
d27729c5bb
refactor(preciseBN): add preciseBN datasets show
2020-04-29 21:05:53 +08:00
liaoxingyu
ec19bcc1d3
style(configs): put all config files together
...
put all config files into one place for easily control,
and add tools for put train_net.py which almost the same in
different projects
2020-04-29 16:18:54 +08:00
liaoxingyu
e38a799b63
fix(engine/defaults): fix precise bn bug
...
fix problem in precise bn, which will not use precise bn datasets, and throw some errors
2020-04-29 16:16:54 +08:00
