Skip to content

Commit 4ba2e16

Browse files
committed
set n_partition to 1 for CVRP
1 parent 6b6446a commit 4ba2e16

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

main.py

+3
Original file line numberDiff line numberDiff line change
@@ -221,6 +221,9 @@ def _eval_dataset(dataset_path, opts, device, revisers):
221221
if opts.width != 1:
222222
opts.width = 1
223223
warnings.warn('Set width to 1 for CVRP!')
224+
if opts.n_partition != 1:
225+
opts.n_partition = 1
226+
warnings.warn('Set n_partition to 1 for CVRP!')
224227
if opts.problem_type == 'pctsp':
225228
if opts.width != 1:
226229
opts.width = 1

0 commit comments

Comments
 (0)