print()s not printing

[nihirv]

Hey,

For some reason I can't seem to run print() functions through the code. I've isolated the problem to this being the case when save_path is included as an argument. For example:

python3 main.py \
--model trs --emb_dim 300 --hidden_dim 300 \
--hop 4 --heads 4 --cuda --batch_size 128 \
--lr 0.001 --pretrain_emb --kl_ceiling 0.48 \
--aux_ceiling 1 --full_kl_step 20000 \
--save_path save/trs_new_bow_batch/ > save/trs_new_bow_batch/out.txt \

Won't print() anything in the code, whereas:

python3 main.py \
--model trs --emb_dim 300 --hidden_dim 300 \
--hop 4 --heads 4 --cuda --batch_size 128 \
--lr 0.001 --pretrain_emb --kl_ceiling 0.48 \
--aux_ceiling 1 --full_kl_step 20000 \
#--save_path save/trs_new_bow_batch/ > save/trs_new_bow_batch/out.txt \

Will run print() correctly.

Has anyone else run into this issue? Is there any stdout/buffer flushing that could be playing this up? I'm on Ubuntu running PyTorch 1.7.1

[nihirv]

Looks like the outputs get written to out.txt - although I can't seem to trace where this writing takes place?