diff --git a/beginner_source/basics/buildmodel_tutorial.py b/beginner_source/basics/buildmodel_tutorial.py index d1501a1fe73..88f86b6dda3 100644 --- a/beginner_source/basics/buildmodel_tutorial.py +++ b/beginner_source/basics/buildmodel_tutorial.py @@ -37,8 +37,8 @@ # `torch.cuda `_ is available, else we # continue to use the CPU. -device = 'cuda' if torch.cuda.is_available() else 'cpu' -print('Using {} device'.format(device)) +device = "cuda" if torch.cuda.is_available() else "cpu" +print(f"Using {device} device") ############################################## # Define the Class diff --git a/beginner_source/basics/quickstart_tutorial.py b/beginner_source/basics/quickstart_tutorial.py index b36979b14da..c5bc0d69e06 100644 --- a/beginner_source/basics/quickstart_tutorial.py +++ b/beginner_source/basics/quickstart_tutorial.py @@ -89,7 +89,7 @@ # Get cpu or gpu device for training. device = "cuda" if torch.cuda.is_available() else "cpu" -print("Using {} device".format(device)) +print(f"Using {device} device") # Define model class NeuralNetwork(nn.Module):