Fix two issues.

The following issues have been fixed.

1. A memory leak issue. If you summarize a tensor that requires gradient, then GC cannot recycle those computed tensors.
2. Previously, you need to pass resuming to the command line interface even if action is set to load.