Closed dancassin closed 1 year ago
I appreciate the notes, I put some of these changes into the PR that updated the readme for v3 of the arXiv paper.
Pytorch/torchvision are such key dependencies that the package assumes researchers already have them installed and does not try to install them for you. I do this to avoid dealing with CUDA version control errors that are gpu-specific.
Followed the instructions provided on the README, and I have some potential changes if agreed upon:
source activate spacetimeformer
should be:conda activate spacetimeformer
in order to run the training commands, it requires a second
cd spacetimeformer
after lines to install requirements.txt in order to run the train.pyAdditon of
torchvision
andopt_einsum
to requirements.txt