Update Actor/actor.py
Browse files- Actor/actor.py +1 -1
Actor/actor.py
CHANGED
@@ -509,7 +509,7 @@ class Actor(nn.Module):
|
|
509 |
|
510 |
if use_time:
|
511 |
#check for arrival times
|
512 |
-
too_far = torch.logical_not(self.
|
513 |
|
514 |
#is depot
|
515 |
a = self.fleet.depot.reshape(self.batch_size, self.num_cars, 1).repeat(1, 1, self.num_nodes)
|
|
|
509 |
|
510 |
if use_time:
|
511 |
#check for arrival times
|
512 |
+
too_far = torch.logical_not(self.check_arival_times())
|
513 |
|
514 |
#is depot
|
515 |
a = self.fleet.depot.reshape(self.batch_size, self.num_cars, 1).repeat(1, 1, self.num_nodes)
|