search

Jannoshh / simple-sam

Sharpness-Aware Minimization for Efficiently Improving Generalization
MIT License
41 stars 9 forks source link

Compatibility change #9

Closed hululuzhu closed 2 years ago

hululuzhu commented 2 years ago

The "*" operator to calculate e_w is not recognized by TF 2.7. Changing to "tf.math.multiply" seemed to solve the problem. Hope it helps.

Jannoshh commented 2 years ago

Thanks!