File tree Expand file tree Collapse file tree 2 files changed +3
-6
lines changed Expand file tree Collapse file tree 2 files changed +3
-6
lines changed Original file line number Diff line number Diff line change 30
30
from transformers import AutoModel
31
31
32
32
import nncf
33
- import nncf .experimental
34
- import nncf .experimental .torch
35
33
from nncf .experimental .torch .fx .quantization .quantize_pt2e import quantize_pt2e
36
34
37
35
@@ -239,7 +237,7 @@ def transform(x):
239
237
action = "store_true" ,
240
238
help = "Enable model validation. --dataset argument is requred for the validation." ,
241
239
)
242
- parser .add_argument ("--dataset" , type = str , help = "Path to the calibration dataset." )
240
+ parser .add_argument ("--dataset" , type = str , help = "Path to the validation dataset." )
243
241
parser .add_argument (
244
242
"--device" ,
245
243
type = str ,
Original file line number Diff line number Diff line change @@ -27,13 +27,12 @@ main() {
27
27
28
28
29
29
# Build the project
30
- cmake --build cmake-openvino-out --target install --config Release -j40
30
+ cmake --build cmake-openvino-out --target install --config Release -j5
31
31
32
32
# # Build example
33
33
local example_dir=examples/openvino
34
34
local example_build_dir=" ${build_dir} /${example_dir} "
35
35
local cmake_prefix_path=" ${PWD} /${build_dir} /lib/cmake/ExecuTorch;${PWD} /${build_dir} /third-party/gflags;"
36
-
37
36
rm -rf " ${example_build_dir} "
38
37
39
38
# # OpenVINO original
@@ -42,7 +41,7 @@ main() {
42
41
-B" ${example_build_dir} " \
43
42
$EXECUTORCH_ROOT /$example_dir
44
43
45
- cmake --build " ${example_build_dir} " -j40
44
+ cmake --build " ${example_build_dir} " -j5
46
45
47
46
# Switch back to the original directory
48
47
cd - > /dev/null
You can’t perform that action at this time.
0 commit comments