federated learning docker
docker build -t <dockerhub-username>/federated .
docker run -it --gpus all <dockerhub-username>/federated /bin/bash
python train.py --dataset_path brain-tumor-train.csv --model_save_path model.h5
python gen_gradients.py --image_dir /brain_tumor_classifier_dataset --gradients_save_path /gradients --model_path model.h5
python update_model.py --model_path model.h5 --saved_gradients gradients --dataset_path brain-tumor-train.csv
python inference.py --model_path model.h5 --image_dir brain_tumor_classifier_dataset --output_json outputs.json
docker push <dockerhub-username>/federated