|
|
|
@ -150,7 +150,7 @@ class Components :
|
|
|
|
else:
|
|
|
|
else:
|
|
|
|
_args['gpu'] = 0
|
|
|
|
_args['gpu'] = 0
|
|
|
|
_args['num_gpu'] = 1
|
|
|
|
_args['num_gpu'] = 1
|
|
|
|
|
|
|
|
os.environ['CUDA_VISIBLE_DEVICES'] = str(args['gpu'])
|
|
|
|
_args['no_value']= args['no_value']
|
|
|
|
_args['no_value']= args['no_value']
|
|
|
|
# MAX_ROWS = args['max_rows'] if 'max_rows' in args else 0
|
|
|
|
# MAX_ROWS = args['max_rows'] if 'max_rows' in args else 0
|
|
|
|
PART_SIZE = int(args['part_size']) if 'part_size' in args else 8
|
|
|
|
PART_SIZE = int(args['part_size']) if 'part_size' in args else 8
|
|
|
|
|
