tensorflowtotal = np.sum([np.prod(v.get_shape().as_list()) for v in tf.trainable_variables()])print('+ Number of params: %.2fM'%(total/1e6))pytorchtotal = sum([param.nelement() for param in model.parameters()])print('+ Number of params: %.2fM'%(total