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 |