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

adapt master to 2.0rc (#347) 

adapt master to 2.0rc
pull/349/head
littletomatodonkey 2020-10-30 00:03:52 +08:00 committed by GitHub
parent 65ec78358f
commit 85cbbf6f3e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
7 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tools/eval.py
View File

@ -24,6 +24,7 @@ sys.path.append(os.path.abspath(os.path.join(__dir__, '../')))
import argparse import argparse
import paddle
import paddle.fluid as fluid import paddle.fluid as fluid
import program import program
@ -83,5 +84,6 @@ def main(args):
if __name__ == '__main__': if __name__ == '__main__':
paddle.enable_static()
args = parse_args() args = parse_args()
main(args) main(args)

3
tools/eval_multi_platform.py
View File

@ -23,7 +23,7 @@ sys.path.append(__dir__)
sys.path.append(os.path.abspath(os.path.join(__dir__, '../'))) sys.path.append(os.path.abspath(os.path.join(__dir__, '../')))
import argparse import argparse
import paddle
import paddle.fluid as fluid import paddle.fluid as fluid
import program import program
@ -77,5 +77,6 @@ def main(args):
if __name__ == '__main__': if __name__ == '__main__':
paddle.enable_static()
args = parse_args() args = parse_args()
main(args) main(args)

2
tools/export_model.py
View File

@ -21,6 +21,7 @@ sys.path.append(os.path.abspath(os.path.join(__dir__, '../')))
import argparse import argparse
from ppcls.modeling import architectures from ppcls.modeling import architectures
import paddle
import paddle.fluid as fluid import paddle.fluid as fluid
@ -81,4 +82,5 @@ def main():
if __name__ == "__main__": if __name__ == "__main__":
paddle.enable_static()
main() main()

2
tools/infer/infer.py
View File

@ -20,6 +20,7 @@ sys.path.append(os.path.abspath(os.path.join(__dir__, '../../')))
import argparse import argparse
import numpy as np import numpy as np
import paddle
import paddle.fluid as fluid import paddle.fluid as fluid
from ppcls.modeling import architectures from ppcls.modeling import architectures
@ -144,4 +145,5 @@ def main():
if __name__ == "__main__": if __name__ == "__main__":
paddle.enable_static()
main() main()

2
tools/infer/py_infer.py
View File

@ -16,6 +16,7 @@ import utils
import argparse import argparse
import numpy as np import numpy as np
import paddle
import paddle.fluid as fluid import paddle.fluid as fluid
@ -99,4 +100,5 @@ def main():
if __name__ == "__main__": if __name__ == "__main__":
paddle.enable_static()
main() main()

2
tools/train.py
View File

@ -25,6 +25,7 @@ sys.path.append(os.path.abspath(os.path.join(__dir__, '../')))
from sys import version_info from sys import version_info
import paddle
import paddle.fluid as fluid import paddle.fluid as fluid
from paddle.fluid.incubate.fleet.base import role_maker from paddle.fluid.incubate.fleet.base import role_maker
from paddle.fluid.incubate.fleet.collective import fleet from paddle.fluid.incubate.fleet.collective import fleet
@ -154,5 +155,6 @@ def main(args):
if __name__ == '__main__': if __name__ == '__main__':
paddle.enable_static()
args = parse_args() args = parse_args()
main(args) main(args)

2
tools/train_multi_platform.py
View File

@ -25,6 +25,7 @@ sys.path.append(os.path.abspath(os.path.join(__dir__, '../')))
from sys import version_info from sys import version_info
import paddle
import paddle.fluid as fluid import paddle.fluid as fluid
from ppcls.data import Reader from ppcls.data import Reader
@ -163,5 +164,6 @@ def main(args):
if __name__ == '__main__': if __name__ == '__main__':
paddle.enable_static()
args = parse_args() args = parse_args()
main(args) main(args)