Article directory
- 1 Preparations
- 2 Tensorboard view training curve online
- 3 Superimpose multiple curves on one graph to display
- 4 Multiple training curves are compared on one graph
- 5 Download the trained model and log checkpoints file
- 6 PatchMatchNet code comment version download (the comment is very detailed, compared with the source code structure, it is more convenient to use)
1 Preparations
1.1 Visualization tool - Tensorboard call
Tensorboard module installation and call reference: PyTorch visualization tool - Tensorboard to view training curves, network structure diagrams, etc. online
This code has been imported, Tensorboard, and called in the corresponding place, see this for details