Skip to content
Permalink
Browse files

Updated jenkins config to install tf-gpu, other minor fixes

  • Loading branch information...
sumitpai committed Apr 11, 2019
1 parent 915fe86 commit 7c4621c7cb6f072cf1ca5bfee2074875db35b3d4
Showing with 3 additions and 6 deletions.
  1. +0 −1 ampligraph/latent_features/models.py
  2. +1 −1 experiments/config.json
  3. +2 −4 jenkins.sh
@@ -447,7 +447,6 @@ def _get_model_loss(self, dataset_iterator):
scores_pos = self._fn(e_s_pos, e_p_pos, e_o_pos)

loss = 0
print(self.loss.get_state('corrupt_side'))

for side in self.loss.get_state('corrupt_side'):
x_neg_tf = generate_corruptions_for_fit(x_pos_tf,
@@ -18,7 +18,7 @@
"FB15K":{
"TRANSE": {"batches_count": 10, "epochs": 4000, "k": 150, "eta": 5, "loss": "pairwise", "loss_params": {"margin": 0.5}, "embedding_model_params": {"norm": 1}, "regularizer": "LP", "regularizer_params": {"lambda": 0.0001, "p":2}, "optimizer": "adam", "optimizer_params": {"lr": 0.0001}, "verbose": false, "seed": 0},
"DISTMULT": {"batches_count":50,"seed":0,"epochs":4000,"k":200,"eta":20,"loss": "self_adversarial","loss_params": {"margin":1},"optimizer":"adam","optimizer_params":{"lr":0.0005},"verbose":false},
"COMPLEX": {"batches_count":100,"seed":0,"epochs":4000,"k":200,"eta":20,"loss": "self_adversarial","loss_params":{"margin":1},"optimizer":"adam","optimizer_params":{"lr":0.0005},"verbose":true},
"COMPLEX": {"batches_count":100,"seed":0,"epochs":4000,"k":200,"eta":20,"loss": "self_adversarial","loss_params":{"margin":1},"optimizer":"adam","optimizer_params":{"lr":0.0005},"verbose":false},
"HOLE": {"batches_count":50,"seed":0,"epochs":4000,"k":200,"eta":20,"loss": "self_adversarial","loss_params": {"margin":1},"optimizer":"adam","optimizer_params":{"lr":0.0005},"verbose":false}
},
"WN18":{
@@ -1,6 +1,4 @@
#Make sure gcc is installed
source /etc/profile
gcc --version
#Make sure gcc is installed
conda --version
pip --version
# Cleanup old env (if any)
@@ -11,7 +9,7 @@ conda create --name ampligraph python=3.6
source activate ampligraph

# Install library
pip install tensorflow==1.12.0
conda install tensorflow-gpu==1.12.0

pip install . -v

0 comments on commit 7c4621c

Please sign in to comment.
You can’t perform that action at this time.