enjoy a pre-trained model after training is done?
erwincoumans opened this issue · 4 comments
Is there a way to enjoy the pre-trained model (weights/checkpoint) after the training is completed?
Usually (TF Agents, OpenAI Baselines, ES Tool etc) there is a 'train' script that saves the weights of the neural network, and a 'enjoy' script that reads those weights and allows to run the pretrained network.
If it is not available, do you have any hints how it can be added?
Hi,
@erwincoumans I would be interested in writing an "enjoy" script as well. Did you manage to do it since you posted this message ?
Thanks!
Léa
I haven't looked into it further. The TensorFlow inspect_checkpoint utility can extract weights from a checkpoint:
https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/tools/inspect_checkpoint.py
Please report back if you make progress.
I've added a note to the README that videos are saved to /tmp as training progresses.
I agree it would be nice to save checkpoints so that training can continue or agents can be viewed. Unfortunately, any free time I find will probably be spent on a new project.