Call read method to load all (param_name, param_val) Returns: a dict of (parameter name, parameter Tensor)
(self)
| 64 | self.snapshot.Write(param_name.encode(), param_val.data) |
| 65 | |
| 66 | def read(self): |
| 67 | '''Call read method to load all (param_name, param_val) |
| 68 | |
| 69 | Returns: |
| 70 | a dict of (parameter name, parameter Tensor) |
| 71 | ''' |
| 72 | params = {} |
| 73 | p = self.snapshot.Read() |
| 74 | for (param_name, param_val) in p: |
| 75 | # print(param_name) |
| 76 | params[param_name] = tensor.from_raw_tensor(param_val) |
| 77 | return params |
no test coverage detected