Sometimes tasks want to capture a dict of data at each timestep.
This commit enables such a feature.
By default, no info will be logged.
Add hooks in default sampler to collect extra info,