Hello Team,
M1 pro van example (PyTorch) example vae: examples/main.py at main · pytorch/examples · GitHub, I use mps:
device = torch.device(“mps” if torch.backends.mps.is_available() else “cpu”)
However the example return non after 10 epoch. For CPU only, every is ok.
Any idea?
Thanks,