8 Commits

Author	SHA1	Message	Date
gaotingquan	162f013ebe	fix: minimize() dont support parameter_list of type dict ... there are diffs that step()+update() and minimize(). this will be fixed in https://github.com/PaddlePaddle/Paddle/pull/53773.	2023-05-25 11:58:05 +08:00
gaotingquan	8b218b01ac	refactor amp auto_cast context manager & loss scaler	2023-05-25 11:58:05 +08:00
gaotingquan	f37cb543b1	rm op black list in amp ... the op flatten_contiguous_range and greater_than has supported amp mode since paddle 2.4	2023-03-29 14:57:02 +08:00
tianyi1997	f313a6d873	Support training without amp	2023-02-28 15:01:21 +08:00
tianyi1997	a9b8432597	fix: wrong base class & simplify train func	2023-02-28 15:01:21 +08:00
tianyi1997	ac466b68da	Create MetaBNNeck	2023-02-28 15:01:21 +08:00
tianyi1997	21e76d08b6	Modify codes based on reviews	2023-02-28 15:01:21 +08:00
tianyi1997	d79fb66e19	Create train func & cfg for MetaBIN	2023-02-28 15:01:21 +08:00