train.py [97:108]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            color_inputs, colors = batch["color_inputs"], batch["colors"]
            lighting_inputs, lightings = batch["lighting_inputs"], batch["lightings"]
            lighting_type_inputs, lighting_types = batch["lighting_type_inputs"], batch["lighting_types"]
            composition_inputs, compositions = batch["composition_inputs"], batch["compositions"]

            losses = []
            for inputs, labels in [
                (color_inputs, colors),
                (lighting_inputs, lightings),
                (lighting_type_inputs, lighting_types),
                (composition_inputs, compositions),
            ]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



train.py [275:286]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
                color_inputs, colors = batch["color_inputs"], batch["colors"]
                lighting_inputs, lightings = batch["lighting_inputs"], batch["lightings"]
                lighting_type_inputs, lighting_types = batch["lighting_type_inputs"], batch["lighting_types"]
                composition_inputs, compositions = batch["composition_inputs"], batch["compositions"]

                losses = []
                for inputs, labels in [
                    (color_inputs, colors),
                    (lighting_inputs, lightings),
                    (lighting_type_inputs, lighting_types),
                    (composition_inputs, compositions),
                ]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -