| 250 | } |
| 251 | |
| 252 | TEST(LoaderCheckpointingTest, TestCheckpoint) { |
| 253 | auto spec = OpSpec("FileReader") |
| 254 | .AddArg("device_id", 0) |
| 255 | .AddArg("max_batch_size", 32) |
| 256 | .AddArg("seed", 123) |
| 257 | .AddArg("checkpointing", true) |
| 258 | .AddArg("pad_last_batch", true); |
| 259 | |
| 260 | TestLoaderCheckpointing(InitLoader<DummyCountingLoader>(spec, 30), 20); |
| 261 | } |
| 262 | |
| 263 | TEST(LoaderCheckpointingTest, TestCheckpointShuffled) { |
| 264 | auto spec = OpSpec("FileReader") |
nothing calls this directly
no test coverage detected