[Feature] [dependency] PyTorch-Optimizers

okbalefthanded / aawedha

Deep Learning toolbox for EEG based Brain-Computer Interface signals decoding and benchmarking

GNU General Public License v3.0

3 stars 0 forks source link

Open okbalefthanded opened 8 months ago

okbalefthanded commented 8 months ago

PyTorch optimizers that are not supported by the official torch implementation are added as separate classes, this behavior makes the codebase heavier, a fix for this is to base the torch optimizers on the PyTorch-Optimizer (as we've done with torchmetrics).