some questions about the code？

[ljjcoder]

It is a great job! But I have some questions:
when I train the the deformation sub-network, it raises a error:
File "onlyBasetwoLoss.py", line 18, in
from datasets import oneShotUnsuperviseCls
ImportError: cannot import name 'oneShotUnsuperviseCls'
I find oneShotUnsuperviseCls do not used in the code, can I delete it?