Open TDteach opened 2 years ago
add the following code to add_argument function of BackdoorAttack class may be a good choice.
group.add_argument('--source_class', type=int, nargs='+',
help='source class(es) of backdoor '
'(default: 1)')
In the init funciton of BackdoorAttack class,
source_class
is a input parameter. However, in the add_argument function of BackdoorAttack class, there is no argument forsource_class
. Also, when typping--help
in the command line, there is nosource_class
argument prompted. Thus, there is no way to setup thesource_class
through command line.