TypeError: argument of type 'method' is not iterable about piqa

[jhyuklee]

python main.py baseline --cuda --mode embed_context --iteration 501 --test_path $SQUAD_DEV_CONTEXT_PATH
python main.py baseline --cuda --mode embed_question --iteration 501 --test_path $SQUAD_DEV_QUESTION_PATH

commands for embedding context and question cause the same error.

[jhyuklee]

state_dict() function seems to be the problem.

https://github.com/uwnlp/piqa/blob/fa113ab2e6cfcec4951d27f6cd4ada6b610c3238/squad/baseline/processor.py#L106-L115

I think it should save self._word2idx_dict instead of self._word2idx.

[jhyuklee]

5ba4f1c solves the problem.
Please check if it's right. @seominjoon

[seominjoon]

Thanks! Will need to put some unit tests in the future...