How to edit the Root?

[qwz3045]

KeyError: 'MixerBlock_0\token_mixing/Dense_0\kernel is not a file in the archive'
While running train.py i get the above error. Any idea why this occurs?

[qwz3045]

I solved the problem.

in modeling.py
befor
`TOK_FC_0 = "token_mixing/Dense_0"
TOK_FC_1 = "token_mixing/Dense_1"
CHA_FC_0 = "channel_mixing/Dense_0"
CHA_FC_1 = "channel_mixing/Dense_1"
PRE_NORM = "LayerNorm_0"
POST_NORM = "LayerNorm_1"

ROOT = f"MixerBlock_{n_block}"
`

after
`TOK_FC_0 = "token_mixing/Dense_0/"
TOK_FC_1 = "token_mixing/Dense_1/"
CHA_FC_0 = "channel_mixing/Dense_0/"
CHA_FC_1 = "channel_mixing/Dense_1/"
PRE_NORM = "LayerNorm_0/"
POST_NORM = "LayerNorm_1/"

ROOT = f"MixerBlock_{n_block}/"
`