Confusion about your code

[daifeng2016]

Hi,
it is interesting to see the result by the KV-net. However, after checking your code, I fail to find any difference between 'unet_model.py' and 'kvnet_model.py', and 'train_unet.py' and 'train_KVnet.py', they are almost the same. I am wondering if there is something wrong about the codes. Could you provide more details about the KV-net?