Skip to content

Don't saved model in train_model.py  #3

@zwd2016

Description

@zwd2016

Hi yoshall,

I'd like to implement you code using sample_data. However, I got some troubles. You can see this information:
total parameters: 554054 ----------epoch 0----------- ----------epoch 1----------- ----------epoch 2----------- ----------epoch 3----------- ----------epoch 4----------- ----------epoch 5----------- ----------epoch 6----------- ----------epoch 7----------- ----------epoch 8----------- ----------epoch 9----------- ----------epoch 10----------- ----------epoch 11----------- ----------epoch 12----------- ----------epoch 13----------- ----------epoch 14----------- stop training !!!
May be not run this code:
if valid_loss < min(valid_losses[:-1]): print('iter {}\tvalid_loss = {:.6f}\tmodel saved!!'.format( iter, valid_loss)) saver.save(sess, model_dir + 'model_{}.ckpt'.format(iter)) saver.save(sess, model_dir + 'final_model.ckpt') else: print('iter {}\tvalid_loss = {:.6f}\t'.format( iter, valid_loss))

So,the previously saved model cannot be loaded when running the test_model.py file.
NotFoundError (see above for traceback): Unsuccessful TensorSliceReader constructor: Failed to find any matching files for ./logs/GeoMAN-12-6-2-128-0.30-0.001/saved_models/final_model.ckpt [[Node: save/RestoreV2 = RestoreV2[dtypes=[DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, ..., DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_INT32, DT_FLOAT], _device="/job:localhost/replica:0/task:0/device:CPU:0"](_arg_save/Const_0_0, save/RestoreV2/tensor_names, save/RestoreV2/shape_and_slices)]]
Look forward to receiving your kind reply soon. Thank you very much.
best wishes,
wendong

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions