Spaces:

alexnasa
/

pixel3dmm

Running on Zero

alexnasa commited on 11 days ago

Commit

5dae933

verified ·

1 Parent(s): d9c00b0

Update src/pixel3dmm/preprocessing/PIPNet/experiments/WFLW/pip_32_16_60_r18_l2_l1_10_1_nb10.py

Files changed (1) hide show

src/pixel3dmm/preprocessing/PIPNet/experiments/WFLW/pip_32_16_60_r18_l2_l1_10_1_nb10.py CHANGED Viewed

@@ -18,4 +18,4 @@ class Config():
         self.save_interval = self.num_epochs
         self.num_nb = 10
         self.use_gpu = True
-        self.gpu_id = 2

         self.save_interval = self.num_epochs
         self.num_nb = 10
         self.use_gpu = True
+        self.gpu_id = 0