Open Duxo opened 1 month ago
Hello @Duxo,
Thank you very much for this very helpful feedback!
Pull request addressing this issue is here: https://github.com/nnaisense/evotorch/pull/111
Feel free to let us know whether or not this pull request correctly addresses the issue you encountered.
Thanks again!
Description:
I am experiencing an issue with the
problem.make_net
method. When I try to generate a trained network using thecenter
parameter from thesearcher.status
, the method fails unless I manually callsqueeze()
on thecenter
parameter.Details:
Code to Reproduce:
Issue:
The code above raises an error when trying to generate a trained network using
problem.make_net(searcher.status["center"])
. However, if I modify the last line to:it works without any issues.
Expected Behavior:
problem.make_net
should handle the shape of thecenter
parameter directly without needing to callsqueeze()
.