|
30 | 30 | model_cpu_tar = file_utils.make_tarfile(mnist_cpu_script,
|
31 | 31 | os.path.join(model_cpu_dir, "model.pth"),
|
32 | 32 | model_cpu_dir,
|
33 |
| - True) |
| 33 | + preserve_script_path=True) |
34 | 34 |
|
35 | 35 | model_cpu_1d_dir = os.path.join(model_cpu_dir, '1d')
|
36 | 36 | mnist_1d_script = os.path.join(model_cpu_1d_dir, code_sub_dir, 'mnist_1d.py')
|
37 | 37 | model_cpu_1d_tar = file_utils.make_tarfile(mnist_1d_script,
|
38 | 38 | os.path.join(model_cpu_1d_dir, "model.pth"),
|
39 | 39 | model_cpu_1d_dir,
|
40 |
| - True) |
| 40 | + preserve_script_path=True) |
41 | 41 |
|
42 | 42 | model_gpu_dir = os.path.join(mnist_path, gpu_sub_dir)
|
43 | 43 | mnist_gpu_script = os.path.join(model_gpu_dir, code_sub_dir, 'mnist.py')
|
44 | 44 | model_gpu_tar = file_utils.make_tarfile(mnist_gpu_script,
|
45 | 45 | os.path.join(model_gpu_dir, "model.pth"),
|
46 | 46 | model_gpu_dir,
|
47 |
| - True) |
| 47 | + preserve_script_path=True) |
48 | 48 |
|
49 | 49 | model_eia_dir = os.path.join(mnist_path, eia_sub_dir)
|
50 | 50 | mnist_eia_script = os.path.join(model_eia_dir, 'mnist.py')
|
|
57 | 57 | os.path.join(model_cpu_dir, "model.pth"),
|
58 | 58 | model_cpu_dir,
|
59 | 59 | "model_call_model_fn_once.tar.gz",
|
60 |
| - True) |
| 60 | + preserve_script_path=True) |
61 | 61 |
|
62 | 62 | ROLE = 'dummy/unused-role'
|
63 | 63 | DEFAULT_TIMEOUT = 20
|
|
0 commit comments