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
......@@ -84,7 +84,7 @@ class Mutator(BaseMutator):
data = dict()
for k, v in self._cache.items():
if torch.is_tensor(v):
v = v.detach().cpu().numpy()
v = v.detach().cpu().numpy().tolist()
if isinstance(v, np.ndarray):
v = v.astype(np.float32).tolist()
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