(
ndarray_base_data_list: List[np.ndarray],
tensors: List[torch.Tensor],
use_cuda: bool,
)
| 26 | |
| 27 | |
| 28 | def assert_ndarray_base_data( |
| 29 | ndarray_base_data_list: List[np.ndarray], |
| 30 | tensors: List[torch.Tensor], |
| 31 | use_cuda: bool, |
| 32 | ): |
| 33 | assert len(ndarray_base_data_list) == len(tensors) |
| 34 | for ndarray_base_data, tensor in zip(ndarray_base_data_list, tensors): |
| 35 | if use_cuda: |
| 36 | tensor = tensor.cpu() |
| 37 | |
| 38 | np.testing.assert_equal(ndarray_base_data, tensor.numpy()) |
| 39 | |
| 40 | |
| 41 | def assert_id_type_feature_data(tensors: List[torch.Tensor], config: dict): |