From: WBC image classification and generative models based on convolutional neural network
Architecture | Learning rate | Decay | Momentum | Dropout | Batch size | Epochs | Hidden unit |
---|---|---|---|---|---|---|---|
W-Net | 0.0001 | Â | Â | 0.6 | 256 | 500 | Â |
W-Net with SVM | 0.0001 | Â | Â | 0.6 | 256 | 500 | Â |
AlexNet | 0.001 | 0.0005 | 0.9 | 0.5 | 128 | 90 | Â |
VGGNet | 0.000001 | Â | Â | 0.5 | 1 | 300 | Â |
ResNet50 | 0.001 | 0.0001 | 0.9 | Â | 32 | 50 | Â |
ResNet18 | 0.001 | 0.0001 | 0.9 | Â | 32 | 50 | Â |
RNN | 0.01 | Â | Â | Â | 64 | Â | 32 |