Fixed the final model saving code
Fixed the final model saving code


Former-commit-id: 5c3c77e1877d56254b1b112a87efa543dab4ce26
Former-commit-id: c798556cf4fc3d92b789e88fa16a0c0a6dff65fa
DeNeutoy/deep_qa,DeNeutoy/deep_qa,allenai/deep_qa,matt-gardner/deep_qa,allenai/deep_qa,matt-gardner/deep_qa