質問編集履歴
1
追記
test
CHANGED
File without changes
|
test
CHANGED
@@ -459,3 +459,65 @@
|
|
459
459
|
これから見るとpythonでGLOBALなオブジェクトを宣言するときはどうも=Noneではまずいらしいですが
|
460
460
|
|
461
461
|
どうすべきか知っている人はいませんか?
|
462
|
+
|
463
|
+
|
464
|
+
|
465
|
+
追記
|
466
|
+
|
467
|
+
GLOBALな変数についてしったのでこう直しても同じでした
|
468
|
+
|
469
|
+
```python
|
470
|
+
|
471
|
+
|
472
|
+
|
473
|
+
def myinit():
|
474
|
+
|
475
|
+
global modelIter
|
476
|
+
|
477
|
+
global device
|
478
|
+
|
479
|
+
global mynet
|
480
|
+
|
481
|
+
|
482
|
+
|
483
|
+
mypath="C:\Users\oakoa\work\fast_neural_style_transfer\model_path"
|
484
|
+
|
485
|
+
|
486
|
+
|
487
|
+
modelPaths = paths.list_files(mypath, validExts=('.pth',))
|
488
|
+
|
489
|
+
modelPaths = sorted(list(modelPaths))
|
490
|
+
|
491
|
+
|
492
|
+
|
493
|
+
modelIter= itertools.cycle(modelPaths)
|
494
|
+
|
495
|
+
print(modelIter)
|
496
|
+
|
497
|
+
# Device
|
498
|
+
|
499
|
+
device = ("cuda" if torch.cuda.is_available() else "cpu")
|
500
|
+
|
501
|
+
# Load Transformer Network
|
502
|
+
|
503
|
+
print("Loading Transformer Network")
|
504
|
+
|
505
|
+
mynet = transformer.TransformerNetwork()
|
506
|
+
|
507
|
+
print(mynet)
|
508
|
+
|
509
|
+
modelPath = next(modelIter)
|
510
|
+
|
511
|
+
mynet.load_state_dict(torch.load(modelPath))
|
512
|
+
|
513
|
+
mynet = mynet.to(device)
|
514
|
+
|
515
|
+
print("Done Loading Transformer Network")
|
516
|
+
|
517
|
+
|
518
|
+
|
519
|
+
|
520
|
+
|
521
|
+
|
522
|
+
|
523
|
+
```
|