When I run:
python3 pretrain.py --config=config_fb15k237.yaml --pretrain_config=TransE
I get this error after the 50th epoch:
Traceback (most recent call last):
File "pretrain.py", line 49, in
gen.pretrain(train_data, corrupter, tester)
File "/KBGAN/trans_e.py", line 80, in pretrain
test_perf = tester()
File "pretrain.py", line 32, in
tester = lambda: gen.test_link(valid_data, n_ent, heads, tails)
File "/KBGAN/base_model.py", line 100, in test_link
for batch_s, batch_r, batch_t in batch_by_size(config().test_batch_size, *test_data):
KeyError: 'test_batch_size'
I get a similar error when I change the option to: --pretrain_config=DistMult. Any idea what might be going wrong? Thanks!
I put test_batch_size in the wrong place in the config file when I refactored the code before release. (this parameter was a hardcoded constant before)
I will check in a fix within a few hours.
Hi Liwei,
When I run: python3 pretrain.py --config=config_fb15k237.yaml --pretrain_config=TransE
I get this error after the 50th epoch: Traceback (most recent call last): File "pretrain.py", line 49, in
gen.pretrain(train_data, corrupter, tester)
File "/KBGAN/trans_e.py", line 80, in pretrain
test_perf = tester()
File "pretrain.py", line 32, in
tester = lambda: gen.test_link(valid_data, n_ent, heads, tails)
File "/KBGAN/base_model.py", line 100, in test_link
for batch_s, batch_r, batch_t in batch_by_size(config().test_batch_size, *test_data):
KeyError: 'test_batch_size'
I get a similar error when I change the option to: --pretrain_config=DistMult. Any idea what might be going wrong? Thanks!