upwindflys
upwindflys
> I guess you should run it with command "horovodrun -np 2 -H localhost:2 python pytorch_mnist.py" ? Thanks.I have modified the spelling error.
I found a solution that reinstall horovod == 0.22.Is there any other tips?
@jinliangwei Thanks for your solution ,but I encountered an oom error,in the meanwhile,free memory is enough,tensorflow and tensor2tensor are both 1.8,Can you offer some suggestions?
@jinliangwei thanks,but that doesn't seem to be this problem,thanks anyway.
@jinliangwei sorry to ask you a question again.What kind of physical environment do you use?such as the cuDNN CUDA version?
Thank you for the reply. But I'm not sure whether this method ```model_to_save = model.module if hasattr(model, 'module') else model``` is a way of ```get_static_torch_model ```? this method's checkpoint is...
@axiniu I met the same problem,Has the problem been solved now?Thanks a lot
@AbelardLiu Thanks a lot,Fix works for me.But why out.set_shape does not work,I feel confused.