added avg total fps (#170)

* added fps as benchmarking stats

* update total avg for fps/rays/s
This commit is contained in:
evonneng authored and GitHub committed 2022-07-18 12:56:32 -07:00
1 parent 5da146e65f
commit 3df29717dd
2 files changed
+16

No files matched your search

+1
View File
@@ -66,6 +66,7 @@ def run_inference(config: DictConfig, local_rank: int = 0, world_size: int = 1)
dataset_inputs_train, _ = setup_dataset_train(config.data, device=device)
_, dataloader_eval = setup_dataset_eval(config.data, test_mode=True, device=device)
graph = setup_graph(config.graph, dataset_inputs_train, device=device)
graph.eval()
# load checkpointed information
_load_checkpoint(config.trainer.resume_train, graph)