diff --git a/README.org b/README.org index a0895f2..1378abf 100644 --- a/README.org +++ b/README.org @@ -17,7 +17,7 @@ Make sure that you have a [[https://www.kaggle.com/docs/api#authentication][Kagg Convert dataset into proper directory format using ~build-dataset.py~, it creates ~/data/train~, ~/data/val~, and ~/data/test/~ (70/10/20 train/validation/test split, though this can be changed in the script) and sub folders in each of those classes so that it can be used by pytorch. ** Train Model -~train.py~ sets the conditions for training the classification model. Trained models and their ".pth" files can be found at our hugging face repository [[https://huggingface.co/jherzstein/HAM10000-Skin-Cancer-Classification/tree/models][Hugging Face repository]] +~train.py~ sets the conditions for training the classification model. Trained models and their ".pth" files can be found at our [[https://huggingface.co/jherzstein/HAM10000-Skin-Cancer-Classification/tree/models][Hugging Face repository]]. *** Arguments Here are some command line arguments for customising the training process: - *-m* (required): Specified model for training