Skip to content

Commit

Permalink
hf README edit
Browse files Browse the repository at this point in the history
  • Loading branch information
jherzstein committed Dec 5, 2024
1 parent 76e2745 commit fc1203f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.org
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit fc1203f

Please sign in to comment.