liaoxingyu
91ff631184
Minor changes
...
Some minor changes, such as class name changing, remove extra blank line, etc.
2021-05-31 17:27:14 +08:00
liaoxingyu
dbf1604231
fix model deploy problems
2021-05-25 15:55:43 +08:00
Sherlock Liao
ff8a958fff
bugfix for plain_train_net.py and lr scheduler step ( #484 )
2021-05-11 15:46:17 +08:00
Xingyu Liao
be0a089e1f
bugfix & merge classification transforms ( #448 )
...
Summary: change heads definition in project and config file, merge classification transforms into default transforms
2021-03-30 15:47:14 +08:00
liaoxingyu
9d83550b67
bugfix
...
Summary: make `deploy` adaptation for other models except repvgg
2021-03-26 10:10:45 +08:00
Xingyu Liao
9b5af4166e
support repvgg ( #429 )
...
Summary:
* support repvgg backbone, and verify the consistency of train mode and eval mode
* onnx export logger style modification
2021-03-11 14:23:24 +08:00
Xingyu Liao
cb7a1cb3e1
update deployment toolchain ( #428 )
...
Summary:
Remove tiny-tensorrt dependency and rewrite a new tensorrt inference api.
In the new version of trt infer, it can pad the input to fixed batch automatically, so you don't need to worry about dynamic batch size.
2021-03-10 16:48:59 +08:00
liaoxingyu
e2a1e14bc3
remove extra info
2021-02-24 14:40:45 +08:00
liaoxingyu
e26182e6ec
make lr warmup by iter
...
Summary: change warmup way by iter not by epoch, which will make it more flexible when training small epochs
2021-01-22 11:17:21 +08:00
liaoxingyu
7e83d3175f
update README.md
...
Summary: add information about fastreid V1.0
2021-01-18 11:44:55 +08:00
liaoxingyu
15e1729a27
update fastreid V1.0
2021-01-18 11:36:38 +08:00
liaoxingyu
7e9a4775da
fixup finetune problem
...
Summary: support finetune from the other model with different number of classes, and simplify calling way (#325 )
close #325
close #325
2020-11-06 15:58:22 +08:00
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
