removed load; iter_train_time includes load time (#351)

This commit is contained in:
evonneng authored and GitHub committed 2022-09-06 16:04:43 -07:00
1 parent 9987c494ab
commit 81fbee213f
3 files changed
+2 -4

No files matched your search

+2 -2
View File
@@ -18,7 +18,7 @@ logging:
LocalWriter:
log_dir: "./"
stats_to_track:
[ITER_LOAD_TIME, ITER_TRAIN_TIME, RAYS_PER_SEC, CURR_TEST_PSNR]
[ITER_TRAIN_TIME, RAYS_PER_SEC, CURR_TEST_PSNR]
max_log_size: 10
```
@@ -46,7 +46,7 @@ WandbWriter:
LocalWriter:
log_dir: "./" # if images are saved, where the images would be saved
stats_to_track:
[ITER_LOAD_TIME, ITER_TRAIN_TIME, RAYS_PER_SEC, CURR_TEST_PSNR]
[ITER_TRAIN_TIME, RAYS_PER_SEC, CURR_TEST_PSNR]
max_log_size: 10
```
-1
View File
@@ -114,7 +114,6 @@ class LocalWriterConfig(InstantiateConfig):
_target: Type = writer.LocalWriter
enable: bool = False
stats_to_track: Tuple[writer.EventName, ...] = (
writer.EventName.ITER_LOAD_TIME,
writer.EventName.ITER_TRAIN_TIME,
writer.EventName.RAYS_PER_SEC,
writer.EventName.CURR_TEST_PSNR,
-1
View File
@@ -45,7 +45,6 @@ class EventName(enum.Enum):
"""Names of possible events that can be logged via Local Writer for convenience.
see config/logging/default_logging.yaml"""
ITER_LOAD_TIME = "Data Load (time)"
ITER_TRAIN_TIME = "Train Iter (time)"
TOTAL_TRAIN_TIME = "Train Total (time)"
ITER_VIS_TIME = "Visualizer Rendering (time)"