Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
wangsen
paddle_dbnet
Commits
6e07bf08
Commit
6e07bf08
authored
Feb 19, 2021
by
LDOUBLEV
Browse files
delete dist when export model
parent
6bafa1ce
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
8 deletions
+1
-8
deploy/slim/prune/export_prune_model.py
deploy/slim/prune/export_prune_model.py
+1
-8
No files found.
deploy/slim/prune/export_prune_model.py
View file @
6e07bf08
...
@@ -25,23 +25,16 @@ sys.path.append(os.path.join(__dir__, '..', '..', '..'))
...
@@ -25,23 +25,16 @@ sys.path.append(os.path.join(__dir__, '..', '..', '..'))
sys
.
path
.
append
(
os
.
path
.
join
(
__dir__
,
'..'
,
'..'
,
'..'
,
'tools'
))
sys
.
path
.
append
(
os
.
path
.
join
(
__dir__
,
'..'
,
'..'
,
'..'
,
'tools'
))
import
paddle
import
paddle
import
paddle.distributed
as
dist
from
ppocr.data
import
build_dataloader
from
ppocr.data
import
build_dataloader
from
ppocr.modeling.architectures
import
build_model
from
ppocr.modeling.architectures
import
build_model
from
ppocr.losses
import
build_loss
from
ppocr.optimizer
import
build_optimizer
from
ppocr.postprocess
import
build_post_process
from
ppocr.postprocess
import
build_post_process
from
ppocr.metrics
import
build_metric
from
ppocr.metrics
import
build_metric
from
ppocr.utils.save_load
import
init_model
from
ppocr.utils.save_load
import
init_model
import
tools.program
as
program
import
tools.program
as
program
dist
.
get_world_size
()
def
main
(
config
,
device
,
logger
,
vdl_writer
):
def
main
(
config
,
device
,
logger
,
vdl_writer
):
# init dist environment
if
config
[
'Global'
][
'distributed'
]:
dist
.
init_parallel_env
()
global_config
=
config
[
'Global'
]
global_config
=
config
[
'Global'
]
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment