Skip to content

Latest commit

 

History

History
20 lines (18 loc) · 862 Bytes

README.md

File metadata and controls

20 lines (18 loc) · 862 Bytes

NeuralTransformer for code completion task

running with float32

# train
CUDA_VISIBALE_DEVICES=0,1,2,3 nohup python -m run.completion.seqrnn.train -f config/py150 > run/completion/seqrnn/config/seqrnn.log 2>&1 &
CUDA_VISIBALE_DEVICES=0,1,2,3 python -m run.completion.seqrnn.train -f config/py150
# eval
CUDA_VISIBALE_DEVICES=0,1,2,3 python -m run.completion.seqrnn.eval -f config/py150
CUDA_VISIBALE_DEVICES=0,1,2,3 python -m run.completion.seqrnn.predictor

running with float16

# train
CUDA_VISIBALE_DEVICES=0,1,2,3 nohup python -m run.completion.seqrnn.train -f config/py150.fp16 > run/completion/seqrnn/config/seqrnn.fp16.log 2>&1 &
CUDA_VISIBALE_DEVICES=0,1,2,3 python -m run.completion.seqrnn.train -f config/py150.fp16
# eval
CUDA_VISIBALE_DEVICES=0,1,2,3 python -m run.completion.seqrnn.eval -f config/py150.fp16