Training ML tasks with in-network aggregation

A distributed PS training architecture with P4 programmable switches accelerating.

Dependency

pytorch needed

sudo apt install libjpeg-dev zlib1g-dev libssl-dev libffi-dev python-dev build-essential libxml2-dev libxslt1-dev

python dependency

  pip3 install pulp numpy tensorboard

cpu only pytorch

pip3 install torch==1.10.0+cpu torchvision==0.11.1+cpu torchaudio==0.10.0+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html

run python3 server.py

Name		Name	Last commit message	Last commit date
Latest commit History 70 Commits
.idea		.idea
data		data
p4switch		p4switch
routing		routing
tests		tests
utils		utils
.gitignore		.gitignore
README.md		README.md
client.py		client.py
config.py		config.py
header_config.py		header_config.py
local_worker_config.json		local_worker_config.json
server.py		server.py
server2_server8_worker_config.json		server2_server8_worker_config.json
server_config.json		server_config.json
worker_config.json		worker_config.json