print cuda device name

app.py

@@ -19,6 +19,7 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 DTYPE = torch.float16
 
 print("DEVICE: ", DEVICE)
+print("CUDA DEVICE NAME: ", torch.cuda.get_device_name(torch.cuda.current_device()))
 
 DEFAULT_FACE_NUMBER = 100000
 MAX_SEED = np.iinfo(np.int32).max