How do I fix your code?

[zlpsls]

I want to test your code on my own datasets. My own datasets have 6 classes. I cannot run your code on my own datasets using your oringinal parameters especially --n and --batach_size.How do I change your code and parameters?