Skip to content

Commit

Permalink
fix docs
Browse files Browse the repository at this point in the history
  • Loading branch information
gonlairo authored and stes committed Jun 29, 2023
1 parent c23a925 commit 663e50d
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion cebra/integrations/sklearn/cebra.py
Original file line number Diff line number Diff line change
Expand Up @@ -1276,7 +1276,6 @@ def to(self, device: Union[str, torch.device]):
>>> cebra_model.fit(dataset)
CEBRA(max_iterations=10)
>>> cebra_model = cebra_model.to("cpu")
>>> cebra_model.is_cuda()
"""

# Check if device is a string or torch.device object
Expand Down

0 comments on commit 663e50d

Please sign in to comment.