-
Notifications
You must be signed in to change notification settings - Fork 1.2k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
AdversarialTrainer parameter name missmatch #2394
Comments
Hi @danielele77 Thank you very much for reporting this issue, we'll fix it as soon as possible. Please let me know if you would be interested to work on a solution for this issue. |
Hi @beat-buesser, thank your response. Due to my time availability, I am unfortunately not able to work on a solution to this problem. |
The problem occures when AdversarialTrainer is used with model, which is trained with Tensorflow/Keras >V2 and fit function is called on instance of AdversarialTrainer.
The problem is in AdversarialTrainer class fit() function in line 262, where is as paramared used nb_epochs, but this is not correct because Keras in older versions used nb_epoch so the Error is raised as you can see on screenshot below.
Keras has back support, but the correct parameter need to be used.
To Reproduce
Steps to reproduce the behavior: (see screenshot below)
Expected behavior
The fit function is exectuted
Screenshots
System information (please complete the following information):
The text was updated successfully, but these errors were encountered: