26 lines
704 B
Python
26 lines
704 B
Python
|
|
|
|
def install_or_import() -> bool:
|
|
try:
|
|
import pip
|
|
try:
|
|
import dadaptation
|
|
except (ModuleNotFoundError, ImportError):
|
|
print("Trying to install dadaptation...")
|
|
pip.main(['install', 'dadaptation'])
|
|
return True
|
|
except (ModuleNotFoundError, ImportError):
|
|
print("Cannot found pip!")
|
|
return False
|
|
return True
|
|
|
|
|
|
def get_dadapt_adam():
|
|
if install_or_import():
|
|
try:
|
|
from dadaptation.dadapt_adam import DAdaptAdam
|
|
return DAdaptAdam
|
|
except (ModuleNotFoundError, ImportError):
|
|
print('Cannot use DAdaptAdam!')
|
|
from torch.optim import AdamW
|
|
return AdamW |