adds default value for is_train
This commit is contained in:
@@ -90,7 +90,7 @@ def get_dataloader(
|
|||||||
num_workers: int,
|
num_workers: int,
|
||||||
anchor_grid: np.ndarray,
|
anchor_grid: np.ndarray,
|
||||||
is_test: bool,
|
is_test: bool,
|
||||||
is_train: bool,
|
is_train: bool = False,
|
||||||
) -> DataLoader:
|
) -> DataLoader:
|
||||||
dataset = MMP_Dataset(
|
dataset = MMP_Dataset(
|
||||||
path_to_data=path_to_data,
|
path_to_data=path_to_data,
|
||||||
|
|||||||
Reference in New Issue
Block a user