xiye17 / torchasn Goto Github PK

When running ./scripts/turk/test.sh checkpoints/turk/pretrained.bin The following error is outputted:

Decoding: 100%|##########| 2500/2500 [06:07<00:00, 6.79it/s]
Traceback (most recent call last):
File "test.py", line 82, in
test(args)
File "test.py", line 72, in test
eq_res = check_equiv(pred_code, gt_code)
File "test.py", line 30, in check_equiv
out = subprocess.check_output(
File "C:\Users\AviVarma\AppData\Local\Programs\Python\Python38\lib\subprocess.py", line 415, in check_output
return run(popenargs, stdout=PIPE, timeout=timeout, check=True,
File "C:\Users\AviVarma\AppData\Local\Programs\Python\Python38\lib\subprocess.py", line 516, in run
raise CalledProcessError(retcode, process.args,
subprocess.CalledProcessError: Command '['java', '-cp', './external/datagen.jar:./external/lib/', '-ea', 'datagen.Main', 'equiv', 'endwith(or(,))', 'endwith(or(,))']' returned non-zero exit status 1.

I am trying to understand the scaled dot product implementation provided in nn.utils.py on line 16. why are you adding a batch_size dimension and what is hidden_size? can you describe the function's input parameters and outputs in more detail please.