(self)
| 224 | transforms(42.1) |
| 225 | |
| 226 | def test_data_loader(self): |
| 227 | xform_1 = mt.Compose([_RandXform()]) |
| 228 | train_ds = Dataset([1], transform=xform_1) |
| 229 | |
| 230 | xform_1.set_random_state(123) |
| 231 | out_1 = train_ds[0] |
| 232 | self.assertAlmostEqual(out_1, 0.2045649) |
| 233 | |
| 234 | set_determinism(seed=123) |
| 235 | train_loader = DataLoader(train_ds, num_workers=0) |
| 236 | out_1 = next(iter(train_loader)) |
| 237 | self.assertAlmostEqual(out_1.cpu().item(), 0.0409280) |
| 238 | |
| 239 | if sys.platform != "win32": # skip multi-worker tests on win32 |
| 240 | train_loader = DataLoader(train_ds, num_workers=1) |
| 241 | out_1 = next(iter(train_loader)) |
| 242 | self.assertAlmostEqual(out_1.cpu().item(), 0.78663897075) |
| 243 | |
| 244 | train_loader = DataLoader(train_ds, num_workers=2) |
| 245 | out_1 = next(iter(train_loader)) |
| 246 | self.assertAlmostEqual(out_1.cpu().item(), 0.785907334) |
| 247 | set_determinism(None) |
| 248 | |
| 249 | def test_data_loader_2(self): |
| 250 | set_determinism(seed=123) |
nothing calls this directly
no test coverage detected