diff --git a/lumin/utils/misc.py b/lumin/utils/misc.py index 8d94eaf..3a9809a 100644 --- a/lumin/utils/misc.py +++ b/lumin/utils/misc.py @@ -12,7 +12,6 @@ __all__ = ['to_np', 'to_device', 'to_tensor', 'str2bool', 'to_binary_class', 'ids2unique', 'FowardHook', 'subsample_df'] device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu') # TODO: make device choosable by user -#device = torch.device('cpu') # TODO: make device choosable by user def to_np(x:Tensor) -> np.ndarray: