Missing Argument and Filtering

[iidsample]

Hi Authors,

The mlp trainer uses a key word argument to write out files.

However this argument is undefined in main_mlp.y
Further we would also like to know how to modify the different sparsity levels for marking activation heads.
Will be great if you can guide us !!