Skip to content

DeepIsolationForest with CUDA #56

@akarpf

Description

@akarpf

How can I train the DeepIsolationForest model on my GPU?

Neither this

from deepod.models.tabular import DeepIsolationForest

clf_dif = DeepIsolationForest(device = 'cuda')

nor this

import torch
from deepod.models.tabular import DeepIsolationForest

device = torch.device("cuda")

clf_dif = DeepIsolationForest(device = device)

works.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions