質問編集履歴
1
補足
test
CHANGED
|
File without changes
|
test
CHANGED
|
@@ -18,6 +18,22 @@
|
|
|
18
18
|
|
|
19
19
|
無限ループに陥っているとかですか？それともちゃんと機械学習できていますか？
|
|
20
20
|
|
|
21
|
+
print文で確認したら、
|
|
22
|
+
|
|
23
|
+
if args.multi_gpu is True:
|
|
24
|
+
|
|
25
|
+
os.environ['CUDA_VISIBLE_DEVICES'] = args.gpu_id # Multi-gpu selector for training
|
|
26
|
+
|
|
27
|
+
net = torch.nn.DataParallel((UNet(residual='pool')).cuda()) # load the network Unet
|
|
28
|
+
|
|
29
|
+
の手前まで実行できていました。
|
|
30
|
+
|
|
31
|
+
しかし、このif,elseはどちらも実行されていません。
|
|
32
|
+
|
|
33
|
+
args.gpu_idに問題があるのかと思うのですが、どう直せばいいですか？
|
|
34
|
+
|
|
35
|
+
使っているGPUは１つ。defaultのままなので、ID=0だと思います。
|
|
36
|
+
|
|
21
37
|
```python
|
|
22
38
|
|
|
23
39
|
import time
|