Source code of torch-pso

S_M · September 25, 2023, 7:47am

Hi all I am trying to implement the PSO as an optimizer in PyTorch. I need to apply some modifications on it. How can I do that?
Thank

ptrblck · September 26, 2023, 4:34am

Are you looking for this repository?

S_M · September 26, 2023, 5:10am

Hi @ptrblck, yes I worked with this library. I tried to classify CIFAR-10. My code was from this reference as you can see the loss function from this reference is CrossEntropy but training error in 10 epochs with default values of the PSO repository was approximately 3000!!! so I tried MSEloss and I saw the error was increased sharply. How can I handle this?