質問編集履歴
2
pwdやlsで実行直下にtrainやtest1があることを確認しました。
test
CHANGED
File without changes
|
test
CHANGED
@@ -1,6 +1,28 @@
|
|
1
|
-
こちらの論文(https://github.com/
|
1
|
+
こちらの論文のギットハブ(https://github.com/jacobgil/pytorch-pruning)の手元の環境で再現を行いたいのですが,どのように実装を行えば良いか分からない状態にあります.
|
2
|
+
|
3
|
+
|
4
|
+
|
2
|
-
|
5
|
+
手元の環境はこちらです。
|
6
|
+
|
7
|
+
|
8
|
+
|
3
|
-
|
9
|
+
```terminal
|
10
|
+
|
11
|
+
$ ls
|
12
|
+
|
13
|
+
README.md data_sh.sh finetune.py test1 tmp train.zip
|
14
|
+
|
15
|
+
__pycache__ dataset.py prune.py test1.zip train
|
16
|
+
|
17
|
+
$ ls train
|
18
|
+
|
19
|
+
cats dogs
|
20
|
+
|
21
|
+
$ ls test1
|
22
|
+
|
23
|
+
cats dogs
|
24
|
+
|
25
|
+
```
|
4
26
|
|
5
27
|
|
6
28
|
|
1
trainやtest1の中身を変えて実装を行いました.
test
CHANGED
File without changes
|
test
CHANGED
@@ -14,13 +14,17 @@
|
|
14
14
|
|
15
15
|
- train
|
16
16
|
|
17
|
+
-- cats ---cat_0.jpg,...,
|
18
|
+
|
17
|
-
|
19
|
+
-- dogs ---dog_0.jpg
|
18
20
|
|
19
21
|
- test1
|
20
22
|
|
23
|
+
-- cats ---5.jpg,...,
|
24
|
+
|
21
|
-
-- 1.jpg,...,
|
25
|
+
-- dogs ---1.jpg,...,
|
26
|
+
|
22
|
-
|
27
|
+
-- 途中
|
23
|
-
|
24
28
|
|
25
29
|
を作って,
|
26
30
|
|
@@ -42,7 +46,7 @@
|
|
42
46
|
|
43
47
|
```terminal
|
44
48
|
|
45
|
-
|
49
|
+
$ python3 finetune.py --train --train_path train --test_path test
|
46
50
|
|
47
51
|
/usr/local/lib/python3.9/site-packages/torchvision/transforms/transforms.py:310: UserWarning: The use of the transforms.Scale transform is deprecated, please use transforms.Resize instead.
|
48
52
|
|
@@ -54,7 +58,7 @@
|
|
54
58
|
|
55
59
|
Traceback (most recent call last):
|
56
60
|
|
57
|
-
File "/Users/
|
61
|
+
File "/Users/***/ronbun/pytorch-pruning/finetune.py", line 270, in <module>
|
58
62
|
|
59
63
|
fine_tuner = PrunningFineTuner_VGG16(args.train_path, args.test_path, model)
|
60
64
|
|