Update README.md
Browse files
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.
|