Remove print

LucasAlegre · Dec 16, 2023 · 507c8b1 · 507c8b1
1 parent af64f1b
commit 507c8b1
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/morl_baselines/multi_policy/envelope/envelope.py b/morl_baselines/multi_policy/envelope/envelope.py
@@ -169,8 +169,6 @@ def __init__(
         self.final_homotopy_lambda = final_homotopy_lambda
         self.homotopy_decay_steps = homotopy_decay_steps
 
-        print("Envelope device: ", self.device)
-
         self.q_net = QNet(self.observation_shape, self.action_dim, self.reward_dim, net_arch=net_arch).to(self.device)
         self.target_q_net = QNet(self.observation_shape, self.action_dim, self.reward_dim, net_arch=net_arch).to(self.device)
         self.target_q_net.load_state_dict(self.q_net.state_dict())