AI-Feynman
AI-Feynman copied to clipboard
Allow custom model when given by the user.
- The user can provide a custom pytorch model.
- Avoid repeating code on a few places.
This looks very reasonable. Any plans to merge this request, @SJ001 ?