Skip to content

GitLab

"src/vscode:/vscode.git/clone" did not exist on "87d8740bf5d8030f9e4e54c9b7e64f353a6f944e"
Unverified Commit 055d7552 authored by Xuesong Wang's avatar Xuesong Wang Committed by GitHub
Browse files

fix: fix the bug in nni/nas/pytorch/mutator, line no. 87 issue#3525 (#3561)

parent ccccfe46
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
nni/nas/pytorch/mutator.py
View file @ 055d7552
...@@ -84,7 +84,7 @@ class Mutator(BaseMutator): ...@@ -84,7 +84,7 @@ class Mutator(BaseMutator):
data = dict() data = dict()
for k, v in self._cache.items(): for k, v in self._cache.items():
if torch.is_tensor(v): if torch.is_tensor(v):
v = v.detach().cpu().numpy() v = v.detach().cpu().numpy().tolist()
if isinstance(v, np.ndarray): if isinstance(v, np.ndarray):
v = v.astype(np.float32).tolist() v = v.astype(np.float32).tolist()
data[k] = v data[k] = v
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment