mirror of
https://github.com/deepfakes/faceswap.git
synced 2026-10-28 21:50:05 +08:00
torzdf
308c5edf39
- Split Loss to own package - Add Focal Frequency Loss (Nvidia only) - Enable GMSD for AMD
14 lines
506 B
Python
14 lines
506 B
Python
#!/usr/bin/env python3
|
|
""" Conditional imports depending on whether the AMD version is installed or not """
|
|
|
|
from lib.utils import get_backend
|
|
|
|
from .normalization import (AdaInstanceNormalization, GroupNormalization, # noqa
|
|
InstanceNormalization, LayerNormalization, RMSNormalization)
|
|
from .loss import losses # noqa
|
|
|
|
if get_backend() == "amd":
|
|
from . import optimizers_plaid as optimizers # noqa
|
|
else:
|
|
from . import optimizers_tf as optimizers #type:ignore # noqa
|