Usage:

First you need to start agents with

CUDA_VISIBLE_DEVICES=0 python agent.py 0

The last 0 stands for the id of the agent. You can spawn multiply agents with each one using one gpu by modifying the visible device id.

Then start the main script by

python simulation_dis.py data.npz 50 40 200

The argvs stands for the name of data, the numbers of individuals in one generation, the numbers of individuals after elimation, the alpha hyperparameter, correspondingly.