minor updates

This commit is contained in:
marcoyang 2024-03-20 12:31:38 +08:00
parent 862ccd700f
commit 18d0333592

View File

@ -393,8 +393,6 @@ def main():
params.update(vars(args)) params.update(vars(args))
device = torch.device("cpu") device = torch.device("cpu")
# if torch.cuda.is_available():
# device = torch.device("cuda", 0)
logging.info(f"device: {device}") logging.info(f"device: {device}")