Skip to content

Commit

Permalink
Update train_em_generalist.py - Fix typo
Browse files Browse the repository at this point in the history
  • Loading branch information
anwai98 authored Oct 25, 2023
1 parent 4e86a21 commit a8df5e8
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
def finetune_em_generalist(args):
"""Example code for finetuning SAM on multiple electron microscopy datasets"""
# override this (below) if you have some more complex set-up and need to specify the exact gpu
device = "cuda" if torch.cuda.is_available() else "gpu"
device = "cuda" if torch.cuda.is_available() else "cpu"

# training settings
model_type = args.model_type
Expand Down

0 comments on commit a8df5e8

Please sign in to comment.