Steven Liu
Hi @JoohyungLee0106 As a similar change is already merged in https://github.com/pytorch/examples/pull/1034 , I'm closing this one. Thanks for your contribution!
I think the print for `--gpu` is useful. To reduce the confusion, we can add some clarification to the print message. Alternative: as DDP is recommended to replace DataParallel, we...
As a similar change is already merged in https://github.com/pytorch/examples/pull/1034 , I'm closing this one.
@mattsta Thanks for your contribution! To move forward, would you mind signing the CLA following the instruction above?
@ArshErgon please rebase and resolve the conflicts.
Closing as it's duplicated with https://github.com/pytorch/examples/pull/1024
Hi @dfarhi , I'm not able to reproduce it with torch==1.12.1+cu102 in Ubuntu 22.04 LTS. Is it still reproducible on your side?
@Deeeerek No. There was a try in https://github.com/pytorch/examples/pull/203 I think it should be easier and cleaner to support fp16 nowadays with amp from pytorch: https://pytorch.org/blog/accelerating-training-on-nvidia-gpus-with-pytorch-automatic-mixed-precision/ Please feel free to contribute...
Looks like we need to include iopath in requirements.txt
I see. I can revert it then.