Implementation of Resnet-RS for training, validation and testing
The Resnet-RS models from: https://github.com/nachiket273/pytorch_resnet_rs
- In the "Initial configuration" part, replace the ImageFolder path with the path of your own custom dataset
- There you can also alter the number of epochs, batch size, k-folds of your preference
- Number of folds (k): 10
- Batch size: 52
- Dropout rate: 0.25
- Loss function: CrossEntropyLoss()
- Number of epochs: 20
- Number of epochs: 200