def as_shared_objects(obj):
if isinstance(obj, tuple):
return tuple(as_shared_objects(x) for x in obj)
elif isinstance(obj, chainer.Link):
return share_params_as_shared_arrays(obj)
elif isinstance(obj, chainer.Optimizer):
return share_states_as_shared_arrays(obj)
elif isinstance(obj, mp.sharedctypes.Synchronized):
return obj
else:
raise ValueError('')
评论列表
文章目录