9 Commits

Author SHA1 Message Date
Tingquan Gao
0efda2c75e Revert "refactor: simpfy engine.train()"
This reverts commit fad5c8e348f748f047508226c3c23335874ef9dc.
2023-03-14 16:47:13 +08:00
Tingquan Gao
5a4ee1aec4 Revert "refactor"
This reverts commit 0e28a39da3952d10efaba71b1329459bb4df9db2.
2023-03-14 16:47:13 +08:00
Tingquan Gao
915dde176a Revert "refactor: rm train and eval from engine"
This reverts commit 5a6fe171a7cfab842adc6a744f11a2e24deb5384.
2023-03-14 16:47:13 +08:00
gaotingquan
5a6fe171a7 refactor: rm train and eval from engine 2023-03-10 16:56:55 +08:00
gaotingquan
0e28a39da3 refactor 2023-03-10 16:56:55 +08:00
gaotingquan
fad5c8e348 refactor: simpfy engine.train()
1. ModelSaver();
2. _build_ema_model();
3. _init_checkpoints();
4. others.
2023-03-10 16:56:55 +08:00
gaotingquan
a38e42f644 refactor: iter_per_epoch -> max_iter 2023-03-10 16:56:55 +08:00
gaotingquan
284e2a6756 refactor: mv all dataloaders to engine.dataloader_dict 2023-03-10 16:56:55 +08:00
gaotingquan
6bed0f5707 refactor: build_train_func & build_eval_func
1. rm engine.device and use paddle.device.get_device() instead;
2. mv some code to build_train_func or build_eval_func to simpfy engine.
2023-03-10 16:56:55 +08:00