Hi, I must say excellent work you have done. But I met some problems while running the code you provided.
First, there is no args.aux in train.py. And there is no testlitstumor in common.py. Maybe there are some mistakes while you updating the code, but I really interesting in your work. If you could solve the bug, I would be really appreciate it.