-
Notifications
You must be signed in to change notification settings - Fork 0
williamlooo/cs182-final
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
Hello, here are instructions for running inference on our project: File Content: README.txt - This file requirements.txt - The python requirments necessary to run this project train_sample.py - A sample training file which trains a simple model on the data, and save the checkpoint to be loaded in the test_submission.py file. conform_validation_set.py - python script needed to reformat the validation set to match what the training script is expecting. test_submission.py - The file which will return an output for every input in the eval.csv eval.csv - An example test file data/get_data.sh - A script which will download the tiny-imagenet data into the data/tiny-imagenet-200 file Note: You should be using Python 3 to run this code. To run inferences first install dependencies through `pip3 install -r requirements.txt`, and then run script: `python3 test_submission.py eval.csv`. Training is relatively straightforward as well. Please do the following from the directory this file is in: `pip3 install -r requirements.txt` if you haven't already. Then run the data downloading script `./data/get_data.sh` Then conform the validation dataset: `python3 conform_validation_set.py`. A folder called val-fixed will be created Finally, run `python3 train_sample_torch.py` Validation of the data is easy too: `python3 evaluation_tools.py`
About
182 compooter wision fwinal pwoject
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published