Skip to content

Commit 21a128d

Browse files
committed
Don't forget to start the loaders
1 parent e4a98f8 commit 21a128d

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/main.cu

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,10 @@ int main(int argc, char* argv[]) {
123123
BatchLoader train_loader {train_files, batch_size};
124124
BatchLoader val_loader {val_files, batch_size};
125125

126-
auto* ds = train_loader.next();
126+
train_loader.start();
127+
val_loader.start();
128+
129+
auto* ds = train_loader.next();
127130

128131
std::cout << ds->positions.size() << std::endl;
129132

0 commit comments

Comments
 (0)