Would it possible for you to release your pre-trained model checkpoint? about wordgcn HOT 6 CLOSED

Hi @Punchwes,
I have uploaded the checkpoint of the pre-trained model. You can download it from the following link:
https://drive.google.com/open?id=1NmYkldyC23fYmyzooojiF3nlnat7rgP9

I have not personally verified whether we can directly restore the checkpoint with the provided code as there have been a few modifications. However, you can access the parameters from the checkpoint.

from wordgcn.

@eyuansu62 If you just want to get parameters or weights you could get to it like below:

ckpt_path = 'best_model/best_int_avg'
reader = tf.train.NewCheckpointReader(ckpt_path)
var_to_shape_map = reader.get_variable_to_shape_map()
embedding = reader.get_tensor('Embed_mat/embed_matrix')

You can get names by var_to_shape_map and use get_tensor to get specific weights.

from wordgcn.

Hi， @Punchwes ， can you use the checkpoint directly? I just restore the model using the checkpoints and it gives some bugs.

from wordgcn.

Hi @eyuansu62 ,

I have not tried to fine-tune the given checkpoint, but I have successfully got access to the parameters in the checkpoint so I guess it should work.

from wordgcn.

@Punchwes i am really a fresh in tensorflow. I just use the saver.restore but it gives some errors. So could you please tell me how to get access to the parameters in the checkpoints?

from wordgcn.

@Punchwes Really thanks!

from wordgcn.