Update actor.py

Author: MichelDeudon

Ptr_Net_TSPTW/actor.py

@@ -206,7 +206,7 @@ def build_optim(self):
         for i in tqdm(range(nb_epoch)): # epoch i
 
             # Get feed_dict
-            input_batch  = training_set.train_batch(actor.batch_size, actor.max_length, actor.input_dimension)
+            input_batch  = training_set.train_batch()
             feed = {actor.input_: input_batch}
             #print(' Input \n', input_batch)
 
@@ -225,4 +225,4 @@ def build_optim(self):
         variables_names = [v.name for v in tf.global_variables() if 'Adam' not in v.name]
         values = sess.run(variables_names)
         for k, v in zip(variables_names, values):
-            print("Variable: ", k, "Shape: ", v.shape)
+            print("Variable: ", k, "Shape: ", v.shape)