diff --git a/models/pose_ae.py b/models/pose_ae.py
index 6fd22ff..f7afc76 100644
--- a/models/pose_ae.py
+++ b/models/pose_ae.py
@@ -1,8 +1,8 @@
 import os
 import sys
 
-os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
-os.environ["CUDA_VISIBLE_DEVICES"] = '0'
+# os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
+# os.environ["CUDA_VISIBLE_DEVICES"] = '0'
 
 import torch
 from torch import nn, Tensor