423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-11-04 18:02:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix eval_batches in V1

Browse Source 
Summary: fix to the D38275943 (597e0259dc).

Reviewed By: bottler

Differential Revision: D38355683

fbshipit-source-id: f326f45279fafa57f24b9211ebd3fda18a518937
This commit is contained in:
Darijan Gudelj 2022-08-02 09:09:47 -07:00 committed by Facebook GitHub Bot
parent b7b188bf54
commit c63ec81750
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pytorch3d/implicitron/dataset/json_index_dataset.py
View File

@ -177,7 +177,9 @@ class JsonIndexDataset(DatasetBase, ReplaceableBase):
raise ValueError(
"Cannot define both eval_batch_index and eval_batches."
)
self.eval_batches = self.seq_frame_index_to_dataset_index()
self.eval_batches = self.seq_frame_index_to_dataset_index(
self.eval_batch_index
)
def is_filtered(self):
"""