我是张力流初学者。所以,请原谅我的问题,如果它是愚蠢的
我检查了一个github代码,用于使用MNIST数据和tensorflow实现CNN。 以下链接:
但是,我需要保存此代码生成的模型,但不知道如何操作,因为此代码不涉及会话的使用,如何在其上进行会话?
非常感谢您的回复。
答案 0 :(得分:0)
链接代码使用import {
MdAutocompleteModule,
MdButtonModule,
MdButtonToggleModule,
MdCardModule,
MdCheckboxModule,
MdChipsModule,
MdDatepickerModule,
MdDialogModule,
MdExpansionModule,
MdFormFieldModule,
MdGridListModule,
MdIconModule,
MdInputModule,
MdListModule,
MdMenuModule,
MdNativeDateModule,
MdPaginatorModule,
MdProgressBarModule,
MdProgressSpinnerModule,
MdRadioModule,
MdRippleModule,
MdSelectModule,
MdSidenavModule,
MdSliderModule,
MdSlideToggleModule,
MdSnackBarModule,
MdSortModule,
MdTableModule,
MdTabsModule,
MdToolbarModule,
MdTooltipModule,
StyleModule,
} from '@angular/material';
import { CdkTableModule } from '@angular/cdk/table';
import { A11yModule } from '@angular/cdk/a11y';
import { BidiModule } from '@angular/cdk/bidi';
import { OverlayModule } from '@angular/cdk/overlay';
import { PlatformModule } from '@angular/cdk/platform';
import { ObserversModule } from '@angular/cdk/observers';
import { PortalModule } from '@angular/cdk/portal';
来训练模型。其documentation包括如何使用tf.estimator.Estimator
保存模型。可以通过export_savedmodel
初始化程序的model_dir
参数指定其位置来导入已保存的模型。