I need a machine learning expert

Hey we want to do federated learning so Our goal is augment our data, create labels and then create a normal cnn with average pooling, 2 CNN layers and Dense with loss Binary crossentrophy and then generating the H5 file for this model then creating two more notebooks named raspberry Pi 1 and 2 then transferring the main H5 file here and training it on some part of unused dataset which we have to seperatebfrom main dataset and generating 2 h5 file from raspberry Pi 1.h5 and raspberry Pi 2.h5 and combining those two h5 files and averaging and creating one super .h5 file and then sending thisbagain back to raspberry Pi 1 and 2 and testing it on unused data