def __init__(self):
super(Agent, self).__init__()
self.p = Parameters()
self.lane_detection_network = lane_detection_network()
self.lane_val = None
self.current_epoch = 0
self.hard_sampling = hard_sampling.hard_sampling()
print("model parameters: ")
print(self.count_parameters(self.lane_detection_network))
def __init__(self, current_epoch=0):
super(Agent, self).__init__()
self.p = Parameters()
self.lane_detection_network = lane_detection_network()
self.setup_optimizer()
self.current_epoch = current_epoch
self.hard_sampling = hard_sampling.hard_sampling()
print("model parameters: ")
print(self.count_parameters(self.lane_detection_network))
def sample_reset(self):
self.hard_sampling = hard_sampling.hard_sampling()