質問編集履歴
2
pwdやlsで実行直下にtrainやtest1があることを確認しました。
title
CHANGED
|
File without changes
|
body
CHANGED
|
@@ -1,6 +1,17 @@
|
|
|
1
|
-
こちらの論文（https://github.com/
|
|
1
|
+
こちらの論文のギットハブ（https://github.com/jacobgil/pytorch-pruning）の手元の環境で再現を行いたいのですが,どのように実装を行えば良いか分からない状態にあります.
|
|
2
2
|
|
|
3
|
+
手元の環境はこちらです。
|
|
3
4
|
|
|
5
|
+
```terminal
|
|
6
|
+
$ ls
|
|
7
|
+
README.md data_sh.sh finetune.py test1 tmp train.zip
|
|
8
|
+
__pycache__ dataset.py prune.py test1.zip train
|
|
9
|
+
$ ls train
|
|
10
|
+
cats dogs
|
|
11
|
+
$ ls test1
|
|
12
|
+
cats dogs
|
|
13
|
+
```
|
|
14
|
+
|
|
4
15
|
おそらく,同じレポジトリ内に
|
|
5
16
|
データをこちら(https://www.kaggle.com/c/dogs-vs-cats/data?select=train.zip)
|
|
6
17
|
からダウンロードし
|
1
trainやtest1の中身を変えて実装を行いました.
title
CHANGED
|
File without changes
|
body
CHANGED
|
@@ -6,10 +6,12 @@
|
|
|
6
6
|
からダウンロードし
|
|
7
7
|
- finetune.py aなど
|
|
8
8
|
- train
|
|
9
|
+
-- cats ---cat_0.jpg,...,
|
|
9
|
-
|
|
10
|
+
-- dogs ---dog_0.jpg
|
|
10
11
|
- test1
|
|
12
|
+
-- cats ---5.jpg,...,
|
|
11
|
-
|
|
13
|
+
-- dogs ---1.jpg,...,
|
|
12
|
-
|
|
14
|
+
-- 途中
|
|
13
15
|
を作って,
|
|
14
16
|
get_parserに定義されているように,
|
|
15
17
|
|
|
@@ -20,13 +22,13 @@
|
|
|
20
22
|
とすれば,良いと思い実装を行いましたが
|
|
21
23
|
|
|
22
24
|
```terminal
|
|
23
|
-
|
|
25
|
+
$ python3 finetune.py --train --train_path train --test_path test
|
|
24
26
|
/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.
|
|
25
27
|
warnings.warn("The use of the transforms.Scale transform is deprecated, " +
|
|
26
28
|
/usr/local/lib/python3.9/site-packages/torchvision/transforms/transforms.py:917: UserWarning: The use of the transforms.RandomSizedCrop transform is deprecated, please use transforms.RandomResizedCrop instead.
|
|
27
29
|
warnings.warn("The use of the transforms.RandomSizedCrop transform is deprecated, " +
|
|
28
30
|
Traceback (most recent call last):
|
|
29
|
-
File "/Users/
|
|
31
|
+
File "/Users/***/ronbun/pytorch-pruning/finetune.py", line 270, in <module>
|
|
30
32
|
fine_tuner = PrunningFineTuner_VGG16(args.train_path, args.test_path, model)
|
|
31
33
|
File "/Users/***/ronbun/pytorch-pruning/finetune.py", line 127, in __init__
|
|
32
34
|
self.train_data_loader = dataset.loader(train_path)
|