The logits layer will only be used, if self.classify is set to True as seen here 3, so I guess this might not be the case.
If so, this layer won’t be used while all other parameters are frozen, which would yield this error.