5 Commits

Author SHA1 Message Date
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