Skip to content
New issue

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

Sign up for GitHub

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

Already on GitHub? Sign in to your account

[refactor] moving memtracer to gemini #801

Merged
merged 17 commits into from
Apr 19, 2022
Merged

[refactor] moving memtracer to gemini #801

Changes from 1 commit
Commits
Show all changes
17 commits
Select commit Hold shift + click to select a range
0ecbf4e
[refactor] moving memtracer to gemini
feifeibear Apr 18, 2022
c441b4d
polish
feifeibear Apr 18, 2022
5a6eea8
fix the stm bug
feifeibear Apr 18, 2022
1411c5e
Merge branch 'main' of https://github.com/hpcaitech/ColossalAI into r…
feifeibear Apr 18, 2022
acb71b8
move gemini unitest to an individual dir
feifeibear Apr 18, 2022
2b23a3e
[log] add tflops logs
feifeibear Apr 18, 2022
d0009f3
polish
feifeibear Apr 18, 2022
031ad4e
polish
feifeibear Apr 18, 2022
16b2677
polish
feifeibear Apr 18, 2022
731ab39
polish
feifeibear Apr 18, 2022
b753c90
polish
feifeibear Apr 18, 2022
ac2f9e3
[log] add tflops logs
feifeibear Apr 18, 2022
7685738
Merge branch 'main' of https://github.com/hpcaitech/ColossalAI into r…
feifeibear Apr 18, 2022
a0109ef
close tests/test_data_pipeline_tensor_parallel/test_cifar_with_data_p…
feifeibear Apr 18, 2022
b46359e
remove imports in class
feifeibear Apr 18, 2022
2de2015
remove some unitests
feifeibear Apr 18, 2022
69a2ead
revert pipeline unittest
feifeibear Apr 18, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
polish
  • Loading branch information
@feifeibear
feifeibear committed Apr 18, 2022
commit d0009f38e59c4abac6c07ea1d4d565c0b10d34aa
2 changes: 1 addition & 1 deletion colossalai/trainer/hooks/_log_hook.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -168,7 +168,7 @@ def _log_by_iter(self, trainer, mode: str):
def _log_by_epoch(self, trainer, mode: str):
for metric_name, metric_calculator in trainer.states['metrics'][mode].items():
if metric_calculator.epoch_only:
val, tflops_value = metric_calculator.get_accumulated_value()
val = metric_calculator.get_accumulated_value()
if self._is_valid_rank_to_log:
self.writer.add_scalar(f'{metric_name}/{mode}', val, trainer.cur_step)

Expand Down