We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 98186fc commit 9c260b1Copy full SHA for 9c260b1
pytorch_optimizer/__init__.py
@@ -245,7 +245,7 @@ def load_optimizer(optimizer: str) -> OPTIMIZER:
245
246
if optimizer.startswith('bnb'):
247
if HAS_BNB and torch.cuda.is_available():
248
- return load_bnb_optimizer(optimizer)
+ return load_bnb_optimizer(optimizer) # pragma: no cover
249
raise ImportError(f'[-] bitsandbytes and CUDA required for bnb optimizers : {optimizer}')
250
if optimizer not in OPTIMIZERS:
251
raise NotImplementedError(f'[-] not implemented optimizer : {optimizer}')
0 commit comments