diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 00000000..7af33347 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,9 @@ +torch +torchvision +numpy +matplotlib +scikit-learn +wandb +tqdm +scipy +h5py6 \ No newline at end of file