INNER CODE UNIT · Python

backend_message

lululxvi/deepxde · deepxde/backend/__init__.py:38

def backend_message(backend_name):
    """Show message about backend.

    Args:
        backend_name: which backend used
    """
    msg = f"Using backend: {backend_name}\nOther supported backends: "
    if backend_name == "tensorflow.compat.v1":
        msg += "tensorflow, pytorch, jax, paddle.\n"
    elif backend_name == "tensorflow":
        msg += "tensorflow.compat.v1, pytorch, jax, paddle.\n"
    elif backend_name == "pytorch":
        msg += "tensorflow.compat.v1, tensorflow, jax, paddle.\n"
    elif backend_name == "jax":
        msg += "tensorflow.compat.v1, tensorflow, pytorch, paddle.\n"
    elif backend_name == "paddle":
        msg += "tensorflow.compat.v1, tensorflow, pytorch, jax.\n"
    msg += "paddle supports more examples now and is recommended."

View source record →