Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

warm_up和EV连用的时候,ckpt里面找不到Ev #985

Open
Confucius-hui opened this issue Apr 1, 2024 · 1 comment
Open

warm_up和EV连用的时候,ckpt里面找不到Ev #985

Confucius-hui opened this issue Apr 1, 2024 · 1 comment
Labels
bug Something isn't working

Comments

@Confucius-hui
Copy link

模型继承训练的代码是这样
`

模型配置文件

warm_start_conf = tf.estimator.WarmStartSettings(ckpt_to_initialize_from=warm_up_model_dir)
model = import_module(self.train_config['model'])
classifier = tf.estimator.Estimator(model_fn=model.model_fn,
                                    params={
                                        'feature_columns': feature_columns,
                                        'config': self.train_config,
                                        'distribution': self.distribution,
                                        'feature_columns_dict': feature_columns_dict
                                        },
                                    warm_start_from=warm_start_conf,
                                    config=run_config)`

在继承训练,load模型文件文件ckpt的时候报错
Pasted Graphic
模型文件里面没有也没有input_layer/1000_embedding/embedding_weights这个key,只有下面的
Pasted Graphic 1
这个该如何解决?

@candyzone
Copy link
Collaborator

对于EV参数,init_from_checkpoint 接口目前支持 assignment_map 传入 scope,不支持传入EV name,用户如果使用Estimator 的warm_start_from 功能暂不支持,需要用户自己调用 init_from_checkpoint 保证EV正常恢复

@candyzone candyzone added the bug Something isn't working label May 13, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
None yet
Development

No branches or pull requests

2 participants