GitLab

The code currently loads the checkpoint and then initializes the variables resulting to random weights. 
Swapping the order fixes the loading checkpoint issue.