mirrors
/
PaddleOCR
mirror of https://github.com/PaddlePaddle/PaddleOCR.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update program.py (#13389) 

refering to-https://www.paddlepaddle.org.cn/documentation/docs/en/api/paddle/incubate/ModelAverage_en.html

it should be paddle.incunate.ModelAverage and paddle.incubate.optimizer.ModelAverage
pull/13406/head
Manik Singh Sarmaal 2024-07-15 19:18:26 +05:30 committed by GitHub
parent 603b3728ac
commit 5347be75f9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tools/program.py
View File

@ -442,7 +442,7 @@ def train(
and dist.get_rank() == 0 and dist.get_rank() == 0
): ):
if model_average: if model_average:
Model_Average = paddle.incubate.optimizer.ModelAverage( Model_Average = paddle.incubate.ModelAverage(
0.15, 0.15,
parameters=model.parameters(), parameters=model.parameters(),
min_average_window=10000, min_average_window=10000,