[TA] Fixed Torch Agent Dict Saving (#2256)
[TA] Fixed Torch Agent Dict Saving (#2256)

* fix saving

* black

facebookresearch/ParlAI,facebookresearch/ParlAI,facebookresearch/ParlAI,facebookresearch/ParlAI,facebookresearch/ParlAI