According to the information provided on OFA github(https://github.com/mit-han-lab/once-for-all/tree/master), you can use your own model, but you need to make sure that the way you construct your model conforms to the OFA reading method.

I would recommend modifying it through MobileNetV3(https://github.com/mit-han-lab/once-for-all/blob/master/ofa/imagenet_classification/networks/mobilenet_v3.py) in OFA github.

Please refer to it. Thank you.