...

2021-04-01 12:43:09 -05:00 · 2021-04-01 12:43:09 -05:00 · 732ccb42e5
parent cf478016b0
commit 732ccb42e5
1 changed files with 2 additions and 0 deletions
--- a/pipeline.py
+++ b/pipeline.py
@ -136,6 +136,8 @@ class Components :
 		# We need to make sure that continuous columns are removed 
 		if x_cols :
 			_args['data'] = df[list(set(df.columns) - set(x_cols))]
+		if 'gpu' in args :
+			_args['gpu'] = args['gpu']
 		data.maker.train(**_args)
 		
 		if 'autopilot' in ( list(args.keys())) :