How can I get model information in customized assessor?

[Roxbili]

I hope to judge whether the model size exceeds a threshold. However, it is hard to pass the model information into Assessor. Has any idea? Thank you!

The assess trial function like this:

from nni.assessor import Assessor, AssessResult

class CustomizedAssessor(Assessor):
    def __init__(self, ...):
        ...

    def assess_trial(self, trial_job_id, trial_history):
        """
        Determines whether a trial should be killed. Must override.
        trial_history: a list of intermediate result objects.
        Returns AssessResult.Good or AssessResult.Bad.
        """
        # you code implement here.
        ...