Skip to content

Commit

Permalink
paddle.save/load support nested structure and layer (#32446)
Browse files Browse the repository at this point in the history
* support save/load binary format tensor

* Fix error when create cudaplace

* Fix error when create cudaplace

* Fix error when create cudaplace

* get devive context from pool.

* move define of 'SerializeToStream' and 'DeserializeFromStream' to 'lod_tensor.cc' and 'selected_rows.cc'.

* support complex object

* improve coverage.

* improve coverage

* improve coverage.

* fix a bug.

* polish API

* save/load program

* paddle.save/load: layer

* deal with conflict

* if PY2, block test_paddle_save_load.TestSaveLoadLayer

* polish code.

* polish code

* edit unnittest

* The condition for object to be identified as state_dict becomes strict

* use 'core._cuda_synchronize'
  • Loading branch information
hbwx24 authored Apr 25, 2021
1 parent 74824fd commit 727b28d
Show file tree
Hide file tree
Showing 3 changed files with 551 additions and 50 deletions.
Loading

0 comments on commit 727b28d

Please sign in to comment.