search

HabanaAI / Model-References

Reference models for Intel(R) Gaudi(R) AI Accelerator
141 stars 67 forks source link

Missing run_lazy_mode option in the argparser in PyTorch's Unet example #20

Closed JoeyTPChou closed 2 years ago

JoeyTPChou commented 2 years ago

In the v. Example: Unet3D in Lazy mode, bf16 mixed precision, Batch Size 2, world-size 8, the run_lazy_mode option is missing. Please check the script utils/utils.py

mandy-li commented 2 years ago

Hi @JoeyTPChou , we fixed the issue. Please check.

JoeyTPChou commented 2 years ago

Hi @mandy-li, thanks for the quick fix. I saw the run_lazy_mode option is removed from the example command line in the README.md. However, they are still in the source code used as a argument without added into the argparser, is that expected?

mandy-li commented 2 years ago

@JoeyTPChou yes, by default, it is true. So you don't need to set it.