Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

GPU-friendly + vectorized pareto ranking #32

Merged
merged 4 commits into from
Nov 25, 2022
Merged

Commits on Nov 25, 2022

  1. - Update _crowding_distance_assignment of evotorch.core function so t…

    …hat it is now vectorized along the pareto front and uses torch tensors only
    
    - Add _compute_pareto_ranks to evotorch.core that computes the rank tensor previously returned by the _pareto_sort_np function, but in a vectorized and GPU-friendly manner
    - Update _pareto_sort function of evotorch.core so that it uses the new _compute_pareto_ranks function
    - Update SolutionBatch.arg_pareto_sort() so that it complies with the update _pareto_sort function (crowdsort_upto argument no longer in use)
    - Add SolutionBatch.compute_pareto_ranks() function which skips the sparse computation of the fronts (useful when not needed)
    - Update SolutionBatch.take_best() function to use SolutionBatch.compute_pareto_ranks() rather than SolutionBatch.arg_pareto_sort()
    - Update evotorch.operators.base.CrossOver._do_tournament so that it uses SolutionBatch.compute_pareto_ranks() rather than SolutionBatch.arg_pareto_sort()
    timothyatkinson authored and Higgcz committed Nov 25, 2022
    Configuration menu
    Copy the full SHA
    4f90135 View commit details
    Browse the repository at this point in the history
  2. Correct comment in evotorch.core._crowding_distance_assignment

    timothyatkinson authored and Higgcz committed Nov 25, 2022
    Configuration menu
    Copy the full SHA
    122c7fb View commit details
    Browse the repository at this point in the history
  3. Add additional coverage of updated pareto_rank functions

    timothyatkinson authored and Higgcz committed Nov 25, 2022
    Configuration menu
    Copy the full SHA
    ebe6028 View commit details
    Browse the repository at this point in the history
  4. Configuration menu
    Copy the full SHA
    606b46a View commit details
    Browse the repository at this point in the history