We try to combine the particle swarm optimization algorithm with the stochastic gradient descent algorithm in deep learning to train and optimize the model, namaly PSOSGD.
pytorch
torchvision
matplotlib
Please see requirements.txt
- PSOSGD Training process
- PSOSGD Testing process
- Model saving
- Plot performance indicators
If you have any questions, please submit an issue or send an email to us. Thanks!
Lanping: [email protected]
Ricky: [email protected]