diff --git a/tools/plantseg/create-config.py b/tools/plantseg/create-config.py index fe82539b..5bb4d5c5 100644 --- a/tools/plantseg/create-config.py +++ b/tools/plantseg/create-config.py @@ -43,7 +43,7 @@ def listify(d, k, sep=',', dtype=float): cfg['preprocessing']['crop_volume'] = '[:,:,:]' cfg['preprocessing']['filter'] = dict(state=False, type='gaussian', filter_param=1.0) - cfg['cnn_prediction']['device'] = 'cuda' + cfg['cnn_prediction']['device'] = 'cpu' cfg['cnn_prediction']['num_workers'] = args.workers cfg['cnn_prediction']['model_update'] = False