Update models.py

models.py

@@ -196,16 +196,15 @@ class ActorCriticSummarisationModel(SummarisationModel):
     def __init__(self, actor_wts, critic_wts, gpu):
 
         super().__init__(gpu)
-
-        self.optimizer = torch.optim.Adam(self.parameters(), lr=0.001)
+        # move to main!, critic wird loss_fn
+        self.critic = Critic(self) # ?
+        critic.load_state_dict(critic_wts)
 
         # actor 
         self.load_state_dict(actor_wts)
 
-        # critic
-        critic = Critic(self) # was übergeben wird? weights? => critic
-        critic.load_state_dict(critic_wts)
-        self.loss_fn = critic
+        
+        
 
     def forward()