from models.lgt_net import LGT_Net