So in my testing I purely used a checkpoint to test the model, using a checkpoint, I can get the tensors and variables via get_collection easily once I restore my model usin