def __call__(self, key, value):
key = key.lstrip('/')
dataset = self.npz[self.path + key]
if isinstance(value, numpy.ndarray):
numpy.copyto(value, dataset)
elif isinstance(value, cuda.ndarray):
value.set(numpy.asarray(dataset))
else:
value = type(value)(numpy.asarray(dataset))
return value
评论列表
文章目录