Source code of torch-pso

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?

Are you looking for this repository?

1 Like

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?