def update()

in model/interpolation_net.py [0:0]

• 13 lines of code
• 4 McCabe index (conditional complexity)

    def update(self, interp_module, i_epoch):
        if i_epoch < self.increase_thresh:  # 0 - 300
            return
        elif i_epoch < self.increase_thresh * 1.5:  # 300 - 450
            num_t = 1
        elif i_epoch < self.increase_thresh * 1.75:  # 450 - 525
            num_t = 3
        else:  # > 525
            num_t = 7

        interp_module.param.num_timesteps = num_t
        print("Set the # of timesteps to ", num_t)

        interp_module.param.lambd_geo = 0
        print("Deactivated the geodesic loss")