Fixing the initialization/loading bug.
The code currently loads the checkpoint and then initializes the variables resulting to random weights. Swapping the order fixes the loading checkpoint issue.
Showing
Please register or sign in to comment