3 Commits

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