Thanks for the code! I’ll have a look at it.
Which script is behaving strangely, train_exclusive.py or train_center_exclusve_filter.py?
Could you tell me which resolution your dataset (casia) has? I would use a random dataset first and check, if there are some obvious mistakes.
Are the default values for the other arguments set, such that the strange behavior is raised?