qthequartermasterman / torch_pso

Particle Swarm Optimization implemented using PyTorch Optimizer API
MIT License
16 stars 1 forks source link

Create a ParamGroups class with mathematical operations #15

Open qthequartermasterman opened 2 years ago

qthequartermasterman commented 2 years ago

Is your feature request related to a problem? Please describe. A clear and concise description of what the problem is. Ex. I'm always frustrated when [...] Currently, we deal with param groups lists and have to manually iterate over groups then by individual params to perform changes param-by-param.

Describe the solution you'd like A clear and concise description of what you want to happen. Wrapping param_groups lists in a class with math operations will allow for cleaner code inside of PSO implementations. This will also allow for more robust unit tests to avoid some of the math bugs I've found while writing PSO variants.

Describe alternatives you've considered A clear and concise description of any alternative solutions or features you've considered.

Additional context Add any other context or screenshots about the feature request here.