search

LouisRouss / Diffusion-Based-Model-for-Colorization

135 stars 11 forks source link

A function is not defined #3

Closed xiaohunshui closed 2 years ago

xiaohunshui commented 2 years ago

There is no definition of partial in trainer.py line 77, can you tell me how to implement it?

xiaohunshui commented 2 years ago

image

LouisRouss commented 2 years ago

Indeed an import was missing:

from functools import partial

xiaohunshui commented 2 years ago

Thank you for your reply, I have solved it with your help!