Skip to content

GitLab

"cacheflow/git@developer.sourcefind.cn:xdb4_94051/vllm.git" did not exist on "e9d3f2ff7772c8efe41dc805cec71c223ec18ec8"
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
Show 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