Update FedProx baseline #3743

Job	Run time
Build, test and upload wheel	1m 30s
Strategy / FedMedian	1m 21s
Framework / bare	1m 41s
Strategy / FedTrimmedAvg	1m 18s
Framework / jax	1m 35s
Strategy / QFedAvg	1m 24s
Framework / pytorch	1m 39s
Strategy / FaultTolerantFedAvg	1m 25s
Framework / tensorflow	1m 41s
Strategy / FedAvgM	1m 22s
Framework / tabnet	1m 31s
Strategy / FedAdam	1m 28s
Framework / opacus	1m 43s
Strategy / FedAdagrad	1m 44s
Framework / pytorch-lightning	1m 42s
Strategy / FedYogi	1m 20s
Framework / mxnet	1m 40s
Framework / scikit-learn	1m 39s
Framework / fastai	1m 26s
Framework / pandas	1m 29s
	30m 38s

Provide feedback