[Train] Disable autofilled metrics

[bveeramani]

Search before asking

• [X] I had searched in the issues and found no similar feature requirement.

Description

Currently, Train reports are autofilled with various metrics.

train.report(epoch=1)
# {'epoch': 1, '_timestamp': 1630716913, '_time_this_iter_s': 0.0039408206939697266, '_training_iteration': 1}

Users might not care about the autofilled metrics. To improve the readability, we could add a feature that disables autofilled metrics.

train.report(epoch=1)
# {'epoch': 1}

Use case

Adding this feature could make reported data easier to work with.

Additionally, disabling autofilled metrics could improve the quality of our tests. Right now, tests are complicated as to handle the autofilled metrics.

assert all(
        all(all(key in worker for key in BASIC_AUTOFILLED_KEYS) for worker in element)
        for element in log
    )

Related issues

No response

Are you willing to submit a PR?

• [ ] Yes I am willing to submit a PR!

[anyscalesam]

Discussed with @matthewdeng; downgrading > might be fixed already Matt can you confirm?