FedScale icon indicating copy to clipboard operation
FedScale copied to clipboard

Published 20 hours ago • verified publisher icon SymbioticLab

Add Tensorflow support

Open mosharaf opened this issue 2 years ago • 2 comments

Ideal solution would be native support. If that's too difficult, as a first step, we should add a tutorial.

mosharaf avatar Jun 07 '22 14:06 mosharaf

Pushed the tensorflow training (#124 Support TF Training). However, there are some loose ends left:

  • Support model testing. This requires us to add tf modules in the current testing functions;

  • Better conversion of data_loaders;

  • Optimize the execution speed. It seems current TF support runs slower than PyTorch.

I will fix these (in order) later.

fanlai0990 avatar Jun 13 '22 07:06 fanlai0990

Great progress. Another TODO would be including appropriate tensorflow in the enviornment.yml file for installation. How do you want to go about it?

mosharaf avatar Jun 13 '22 18:06 mosharaf