wang zhenyang

Results 1 comments of wang zhenyang

#### 1、from oneflow.cuda import amp报错:oneflow.cuda里没有amp模块 torch.cuda.amp自动混合精度训练 —— 节省显存并加快推理速度 解决方法:直接将torch里的amp模块复制到本地import #### 2、在套用torch_distributed_zero_first函数时,会报错:oneflow.distributed没有barrier torch_distributed_zero_first函数是用来处理模型进行分布式训练时的同步问题 解决方法:直接注释掉与torch_distributed_zero_first有关代码,不采用分布式训练同步 #### 3、oneflow.load加载pt模型参数时,报错:load() got an unexpected keyword argument 'map_location' 没有map_location这个参数 目前解决方法是去掉map_loaction,但是再运行会显示模型加载地址找不到的问题