arch
|
fix head_init_scale
|
2023-05-26 15:40:48 +08:00 |
configs
|
add amp args, use_amp=False
|
2023-05-29 19:52:09 +08:00 |
data
|
getargspec -> getfullargspec
|
2023-05-17 15:19:13 +08:00 |
engine
|
fix _init_amp when export
|
2023-05-29 19:52:09 +08:00 |
loss
|
Revert "refactor: simplify engine"
|
2023-03-14 16:47:13 +08:00 |
metric
|
Revert "refactor: simplify engine"
|
2023-03-14 16:47:13 +08:00 |
static
|
fix data dtype for amp training
|
2023-04-26 18:40:20 +08:00 |
utils
|
fix ema: set_value() -> paddle.assign()
|
2023-05-26 15:40:48 +08:00 |