我添加了一個自定義的操作並使用ResourceMgr創建ResourceBase的子類來存儲一些狀態,以及如何將狀態寫入檢查點文件。如何保存和恢復ResourceBase?
class FeatureTransformMap : public ResourceBase {
public:
FeatureTransformMap(int32_t max_feature_count) : max_feature_count_(max_feature_count), feature_index_(0) {
cout<<"Max feature count is:"<<max_feature_count_<<endl;
}
string DebugString() {return "FeatureTransformMap";}
int32_t GetFeatureIndex(const string& feature) {
{
mutex_lock l(mu_);
feature_index_ += 1;
}
return feature_index_;
}
private:
tensorflow::mutex mu_;
uint32_t feature_index_ GUARDED_BY(mu_);
const uint32_t max_feature_count_;
};
在上面的代碼中,我怎麼能寫feature_index_檢查點文件。