Skip to content

Latest commit

 

History

History
13 lines (12 loc) · 745 Bytes

Parameter.md

File metadata and controls

13 lines (12 loc) · 745 Bytes
* Number of training samples = 429 {smaller batch}
* Number of training samples = 4270    {larger batch}
* Number of silos = 2 
* Test Batch Size = 8 ( Reduced largely due to computational restrictions ) 
* Train Batch Size = 4 
* lr = 0.001
* Optimizer = SGD 
Note : DICOM images in the linked dataset are of the following dimension - [1024,1024]. In the provided code they were read as [3,128,128]. We could not increase the dimensions in which we read the images beyond this due to computational restricitions. Please note that the main aim of the paper is not to provide a state of the art model to detect pneumonia using Chest-X Rays but rather demonstrate the efficacy of the cross-silo FL setup with the given use case.