zhangcen456 / IB

7 stars 1 forks source link

Reproducibility Issue #2

Closed zeroshu closed 2 years ago

zeroshu commented 2 years ago

Can you briefly introduce the running instructions? I can't reproduce your results.

zhangcen456 commented 2 years ago

you can change the dimensionality of hidden layers using "-hd" or "--hidden_size" and change beta using '-b' or '--beta'. For example, for SST-2 dataset you may run "python model.py -h 150 -b 1.5 -as 0" ("-as 0" for not applying adversarial training). The hyper-parameter we used can be found in the paper.