update

RutgersDM · Aug 20, 2022 · 77bb40a · 77bb40a
1 parent 33dfbae
commit 77bb40a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/configs/fb15k-237.sh b/configs/fb15k-237.sh
@@ -2,13 +2,13 @@
 
 data_input_dir="datasets/data_preprocessed/FB15K-237/"
 vocab_dir="datasets/data_preprocessed/FB15K-237/vocab"
-total_iterations=5000
+total_iterations=10000
 path_length=3
 hidden_size=50
 embedding_size=50
-batch_size=256
-beta=0.1
-Lambda=0.1
+batch_size=512
+beta=0.14
+Lambda=0.14
 use_cluster_embeddings=0
 use_entity_embeddings=0
 train_entity_embeddings=0

diff --git a/trainer.py b/trainer.py
@@ -353,7 +353,7 @@ def train(self):
 						format(self.batch_counter, np.sum(entity_rewards), e_avg_reward + c_avg_reward, num_ep_correct,
 							   (num_ep_correct / self.batch_size), train_loss))
 
-			if self.batch_counter % self.eval_every == 0 or np.sum(entity_rewards) >= 1800:
+			if self.batch_counter % self.eval_every == 0: #or np.sum(entity_rewards) >= 1800:
 
 				self.test_rollouts = 100
 				self.test_environment = self.test_test_environment