Table 4 Technical information for training.
Parameter | Value |
|---|---|
Framework | PyTorch |
GPU | NVIDIA GeForce GTX 1080 Ti |
Training time | 3.53 h |
Optimizer | Adaptive moment estimation (Adam) |
Learning rate | 0.00008 |
Loss function | Mean squared error loss (MSELoss) |
Max Epochs | 4 |
Batch Size | 32 |