src/utils.py

def loss_fn(images):
    return -images.median() / 3


concept_styles = {
    "Coffee Machine": "coffeemachine.bin",
    "College Style": "college_style.bin",
    "Cube": "cube.bin",
    "Jerry Mouse": "jerrymouse",
    "Zero": "zero.bin",
}