How to use it in other .py file?

[EEstq]

Hi, I use your code and it run successfully.But I can't get the result when I use it in its sub-module
it prints nothing when the function is called
e.g.

def dis_update(self, x_a, x_b, hyperparameters):
        gpu_tracker.track()
        self.dis_opt.zero_grad()
        x_a_1,x_a_2,h_a, n_a = self.gen_a.encode(x_a)
        x_b_1,x_b_2,h_b, n_b = self.gen_b.encode(x_b)
        # decode (cross domain)
        x_ba = self.gen_a.decode(x_b_1,x_b_2,h_b + n_b)
        x_ab = self.gen_b.decode(x_a_1,x_a_2,h_a + n_a)
        self.loss_dis_total = ......
        self.loss_dis_total.backward()
        self.dis_opt.step()
        gpu_tracker.track()

Can you help me with the problem?