Closed VocAddict closed 4 years ago
Added torch and torchvision dependency checks and download options to main.py, as well as dependency checks on the scripts that use them. Minor updates to verify user input, more options for user exiting scripts, and renaming script files as well.
Commented out trainer and sampler files from the main script along with its dependency checks until further notice. I was unable to get torch and torchvision to install on a virtual container and as such unable to reproduce why it works on my system despite everything seeming to look the same on both ends.
I've tested it in Anaconda and trainer and sampler work there as expected. As such the best course of action at this point is to run the entire project under conda. I've tested the other scripts as well and it works the same way it does under standard Python.
Somehow the file that got pushed, and what I had on my local system did not match so here's an update with sample.py running as it should. Please confirm before merging because I don't trust my computer right now.