devadethanr commited on
Commit
71669e9
·
verified ·
1 Parent(s): 8334394

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -6
README.md CHANGED
@@ -19,13 +19,13 @@ SqueezeNet: Very small but less accurate than the other options.
19
  Ultimately, EfficientNet-B0 was chosen due to its balance of accuracy and efficiency. While slightly larger than MobileNetV2, it was expected to fit within the free tier's resource limits and potentially provide better performance for this medical imaging task.
20
  Link : Efficientnet-b0
21
 
22
- #### Hyperparameter Choices:
23
 
24
- per_device_train_batch_size=32
25
- num_train_epochs=25
26
- fp16=True : Mixed precision training is enabled to leverage the GPU's Tensor Cores (if available) and speed up training.
27
- load_best_model_at_end=True : This ensures that we use the best performing model
28
- push_to_hub=True & hub_model_id & hub_strategy = "every_save"
29
  - 4. Preprocessing
30
  The dataset's images were already in a usable format (PIL). The preprocessing steps involved:
31
  Splitting: The dataset was split into training (80%) and validation (20%) sets.
 
19
  Ultimately, EfficientNet-B0 was chosen due to its balance of accuracy and efficiency. While slightly larger than MobileNetV2, it was expected to fit within the free tier's resource limits and potentially provide better performance for this medical imaging task.
20
  Link : Efficientnet-b0
21
 
22
+ - #### Hyperparameter Choices:
23
 
24
+ - - per_device_train_batch_size=32
25
+ - - num_train_epochs=25
26
+ - - fp16=True : Mixed precision training is enabled to leverage the GPU's Tensor Cores (if available) and speed up training.
27
+ - - load_best_model_at_end=True : This ensures that we use the best performing model
28
+ - - push_to_hub=True & hub_model_id & hub_strategy = "every_save"
29
  - 4. Preprocessing
30
  The dataset's images were already in a usable format (PIL). The preprocessing steps involved:
31
  Splitting: The dataset was split into training (80%) and validation (20%) sets.