Skip to content

Commit

Permalink
refine log
Browse files Browse the repository at this point in the history
  • Loading branch information
eunwoosh committed Aug 1, 2023
1 parent 167cf27 commit c86ba4d
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,7 @@ def adapt_batch_size(train_func: Callable, cfg, datasets: List, validate: bool =
"""

if not cuda_available():
logger.warning("Auto-adaptive batch size feature can't works without a CUDA device. It is skipped.")
logger.warning("Skip Auto-adaptive batch size: CUDA should be available, but it ins't.")
return

def train_func_single_iter(batch_size):
Expand Down

0 comments on commit c86ba4d

Please sign in to comment.