Skip to content

Commit

Permalink
Formatting source.
Browse files Browse the repository at this point in the history
  • Loading branch information
YukioOobuchi committed Jun 18, 2018
1 parent beb311a commit 1b28d8b
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 1 deletion.
1 change: 1 addition & 0 deletions python/src/nnabla/utils/cli/cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -78,6 +78,7 @@ def main():

args.func(args)


if __name__ == '__main__':
import six.moves._thread as thread
import threading
Expand Down
3 changes: 2 additions & 1 deletion python/src/nnabla/utils/load.py
Original file line number Diff line number Diff line change
Expand Up @@ -697,7 +697,8 @@ class Info:
text_format.Merge(f.read(), proto)
if len(proto.parameter) > 0:
if not exclude_parameter:
nn.load_parameters(os.path.join(tmpdir, name))
nn.load_parameters(
os.path.join(tmpdir, name))
elif ext in ['.protobuf', '.h5']:
nnp.extract(name, tmpdir)
if not exclude_parameter:
Expand Down

0 comments on commit 1b28d8b

Please sign in to comment.