Table 2.
Parameters used for model training.
| Parameters | Settings |
|---|---|
| Hardware | GPU, CPU |
| Framework | PyTorch |
| Optimizer | Adam |
| Learning rate | 1e-4 |
| Batch size | 8 |
| Loss function | CrossEntropyLoss |
| Epochs | 150 |
Official websites use .gov
A
.gov website belongs to an official
government organization in the United States.
Secure .gov websites use HTTPS
A lock (
) or https:// means you've safely
connected to the .gov website. Share sensitive
information only on official, secure websites.