MSAI Group 6

Zhengxiao Han, Zhengyang (Kris) Weng, Ben Benyamin

Installation

pip install -r requirements.txt

In data_generation:

Terminal 1: Launch Tensorboard

tensorboard --logdir=runs

Terminal 2: Start Training

python preliminary/train.py

Then open http://localhost:6006/ in your web browser. It will take a long time to load data before plotting in web browser.

Name		Name	Last commit message	Last commit date
Latest commit History 31 Commits
data_generation		data_generation
preliminary		preliminary
proposal		proposal
.gitignore		.gitignore
README.md		README.md
knn.py		knn.py
model.py		model.py
model_6d.py		model_6d.py
model_han.py		model_han.py
requirements.txt		requirements.txt
requirements_cuda121.txt		requirements_cuda121.txt
test.py		test.py
train.py		train.py
train_6d.py		train_6d.py