Table 2.
Parameters | Learning from scratch | Transfer learning | |
---|---|---|---|
Training fully connected layers | Fine-tuning | ||
Training algorithm | SGD | RMSP | SGD |
Learning rate | 0.01 | 0.01 | 0.0001 |
Batch size | 32 | ||
Early stopping | 10 epochs |
Parameters | Learning from scratch | Transfer learning | |
---|---|---|---|
Training fully connected layers | Fine-tuning | ||
Training algorithm | SGD | RMSP | SGD |
Learning rate | 0.01 | 0.01 | 0.0001 |
Batch size | 32 | ||
Early stopping | 10 epochs |