fix print format warning for size_t vs int

staar · May 21, 2024 · 967420d · 967420d
1 parent 587506d
commit 967420d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/train_gpt2.c b/train_gpt2.c
@@ -1041,8 +1041,8 @@ int main() {
     DataLoader train_loader, val_loader;
     dataloader_init(&train_loader, train_tokens, B, T, 0, 1);
     dataloader_init(&val_loader, val_tokens, B, T, 0, 1);
-    printf("train dataset num_batches: %d\n", train_loader.num_batches);
-    printf("val dataset num_batches: %d\n", val_loader.num_batches);
+    printf("train dataset num_batches: %zu\n", train_loader.num_batches);
+    printf("val dataset num_batches: %zu\n", val_loader.num_batches);
     int val_num_batches = 5;
 
     // build the Tokenizer