From d011b419354e2b152208c91b73e3fd6c5fcfe4e0 Mon Sep 17 00:00:00 2001 From: Tingfeng Lan Date: Mon, 31 Jul 2023 07:49:18 +0000 Subject: [PATCH] Formatting code style. --- model_zoo/pytorch/nanogpt/train.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/model_zoo/pytorch/nanogpt/train.py b/model_zoo/pytorch/nanogpt/train.py index 8660f9bba..c51c9a568 100644 --- a/model_zoo/pytorch/nanogpt/train.py +++ b/model_zoo/pytorch/nanogpt/train.py @@ -243,8 +243,13 @@ def train(): # Training loop X, Y = get_batch( - "train", train_data=train_data, val_data=val_data, device=device, - device_type=device_type, batch_size=batch_size, block_size=block_size + "train", + train_data=train_data, + val_data=val_data, + device=device, + device_type=device_type, + batch_size=batch_size, + block_size=block_size, ) # Fetch the very first batch total_time = 0.0 local_iter_num = 0 # Number of iterations in the lifetime of this process