我的网络参数已经在gpu上初始化了,但是我想把它们移到cpu上。我试着把网络作为一个整体复制:# obviously doesn't workfor param in net.collect_params(): data = data.as_in_context(mx.cpu())
# does notha
当我加载一个pickle文件到CPU时,我遇到了一个问题。我在网上搜索了一下,他们说我需要添加map_location参数。但是,在我添加了这个参数之后,问题仍然存在。torch.__version__Attempting to deserializeIf you are running
on a <