tensorflow==2.4.0 tensorflow-model-optimization==0.5.0 numpy==1.19.5