Commit Graph

9 Commits (0efda2c75e1cc9c1d1a6168cc7bcf0506236ae93)

Author SHA1 Message Date
Tingquan Gao 0efda2c75e Revert "refactor: simpfy engine.train()"
This reverts commit fad5c8e348.
2023-03-14 16:47:13 +08:00
Tingquan Gao 5a4ee1aec4 Revert "refactor"
This reverts commit 0e28a39da3.
2023-03-14 16:47:13 +08:00
Tingquan Gao 915dde176a Revert "refactor: rm train and eval from engine"
This reverts commit 5a6fe171a7.
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