-
Notifications
You must be signed in to change notification settings - Fork 37
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
yolov3 问题 #356
Comments
1、from oneflow.cuda import amp报错:oneflow.cuda里没有amp模块torch.cuda.amp自动混合精度训练 —— 节省显存并加快推理速度 2、在套用torch_distributed_zero_first函数时,会报错:oneflow.distributed没有barriertorch_distributed_zero_first函数是用来处理模型进行分布式训练时的同步问题 3、oneflow.load加载pt模型参数时,报错:load() got an unexpected keyword argument 'map_location'没有map_location这个参数 |
第一点,我们的amp暂时在Graph层支持(即静态图),Eager动态图还不支持AMP,所以直接复制amp模块import我大概率估计是跑不起来 |
oneflow 里如果没有对应的模块,可能是没实现,或者机制不同,可以注释掉相关代码,或者换个方式实现,不要直接硬搬过来。
问题再描述清楚一些,具体这个函数内是做了啥,把具体代码段贴一下,方便他同事解答问题。
oneflow.load 不能直接加载torch的模型 |
No description provided.
The text was updated successfully, but these errors were encountered: