Particle Swarm Optimization for Dataparallel Neural Nets.
This is the repository for my Master's thesis.
To run the script on the server:
- Clone the repository via: "git clone https://github.com/masterbaer/PSO_for_DPNN.git"
- Navigate to the folder with the .sh files.
- Create a virtual environment with "python -m venv venv".
- Upgrade pip twice "pip install --upgrade pip" "pip install --upgrade pip"
- Install some python packages: "pip install numpy torch torchvision torch-pso matplotlib scikit-learn mpi4py torch-pruning" For mpi4py to install successfully, you need to load the module first via "module load mpi/openmpi/4.1".
- Close the virtual environment again "deactivate".
Now you can run the scripts with e.g. "sbatch submit_alexnet_sequetial.sh".