Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Modify the log for hooks_in_train/test (#178 ) #181

Merged
merged 3 commits into from
Jun 23, 2022

Conversation

DavdGao
Copy link
Collaborator

@DavdGao DavdGao commented Jun 22, 2022

To solve #178 , the current log of hooks_in_train and hooks_in_eval is as follows

2022-06-22 02:57:53,122 (trainer:303) INFO: After register default hooks,
	the hooks_in_train is:
	{
	  "on_fit_start": [
	    "_hook_on_fit_start_init",
	    "_hook_on_fit_start_calculate_model_size"
	  ],
	  "on_epoch_start": [
	    "_hook_on_epoch_start"
	  ],
	  "on_batch_start": [
	    "_hook_on_batch_start_init"
	  ],
	  "on_batch_forward": [
	    "_hook_on_batch_forward",
	    "_hook_on_batch_forward_regularizer",
	    "_hook_on_batch_forward_flop_count"
	  ],
	  "on_batch_backward": [
	    "_hook_on_batch_backward"
	  ],
	  "on_batch_end": [
	    "_hook_on_batch_end"
	  ],
	  "on_fit_end": [
	    "_hook_on_fit_end"
	  ]
	};
	the hooks_in_eval is:
	{
	  "on_fit_start": [
	    "_hook_on_fit_start_init"
	  ],
	  "on_epoch_start": [
	    "_hook_on_epoch_start"
	  ],
	  "on_batch_start": [
	    "_hook_on_batch_start_init"
	  ],
	  "on_batch_forward": [
	    "_hook_on_batch_forward"
	  ],
	  "on_batch_end": [
	    "_hook_on_batch_end"
	  ],
	  "on_fit_end": [
	    "_hook_on_fit_end"
	  ]
	}

@DavdGao DavdGao requested review from yxdyc and xieyxclack June 22, 2022 03:03
@DavdGao DavdGao changed the title Modify the log for hooks_in_train/test Modify the log for hooks_in_train/test (#178 ) Jun 22, 2022
@xieyxclack xieyxclack linked an issue Jun 22, 2022 that may be closed by this pull request
xieyxclack
xieyxclack previously approved these changes Jun 22, 2022
Copy link
Collaborator

@xieyxclack xieyxclack left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@xieyxclack xieyxclack self-requested a review June 23, 2022 02:28
@xieyxclack xieyxclack merged commit bf5c42e into alibaba:master Jun 23, 2022
# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Improve the current logging for the report of routines.
2 participants