Has anyone managed to solve the error (see below) that arises when attempting to use aug_transforms technique on macos m1?
NotImplementedError: The operator 'aten::_linalg_solve_ex.result' is not currently implemented for the MPS device. If you want this op to be added in priority during the prototype phase of this feature, please comment on https://github.com/pytorch/pytorch/issues/77764. As a temporary fix, you can set the environment variable `PYTORCH_ENABLE_MPS_FALLBACK=1` to use the CPU as a fallback for this op. WARNING: this will be slower than running natively on MPS.
Update: using chatgpt4 I learned that adding below adjustment helps a little bit:
where custom_aug_transforms is
from fastai.vision.augment import * custom_aug_transforms = [RandomResizedCrop(128, min_scale=0.35), Flip(), Brightness(), Contrast(), Rotate(max_deg=10.0)]
I am not sure impact of
custom_aug_transforms is equally good as
aug_transforms but seems not too bad. In the meantime please visit github issue to implement aten::_linalg_solve_ex.result operator and vote for it to increase its priority