v0.5.1

Fixed issues

• Fixed issue with FEMNIST (#classes 64 -> 62) and an inefficiency on DataSplitter.assign
• Fixed issue related to batch normalization layers

New features

• Added option to unload models on disk to save memory (inmemory)
• Added fine_tuning_epochs to clients so the fine-tuning is now possible with a custom number of local epochs
• Added output_size argument to some nets
• Added the possibility to pass arguments to build the model through the configuration (net_args option)
• Added Saving options
• Added FL algorithm DPFedAVG