How to edit the Root?
qwz3045 opened this issue · comments
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?
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}/"
`