search

cleverhans-lab / cleverhans

An adversarial example library for constructing attacks, building defenses, and benchmarking both
MIT License
6.15k stars 1.39k forks source link

Custom loss in MIM in TF2 #1199

Closed YutianPangASU closed 3 years ago

YutianPangASU commented 3 years ago

Added softmax cross-entropy loss as in the original paper.

YutianPangASU commented 3 years ago

This pull request is raised to address the issue in #1191.

alkaet commented 3 years ago

Thanks a lot for PR. Can you format the code using Black?

YutianPangASU commented 3 years ago

Thanks a lot for PR. Can you format the code using Black?

No problem. I was having some issues formatting the code so I edited it manually. Please let me know if there is any other issue.

alkaet commented 3 years ago

I'll have another look and test it once it is formatted in Black.

YutianPangASU commented 3 years ago

I'll have another look and test it once it is formatted in Black.

Yes, please have a look when you are available.

alkaet commented 3 years ago

lgtm. Thanks!