Enable basic logging in CIFAR-10 examples in Shampoo

Summary: See title.

Reviewed By: jialun-zhang

Differential Revision: D88899177

fbshipit-source-id: 9ff0c8092d9ddd4708ac332b41beab80e79f7922

Author: Michael Shi

distributed_shampoo/examples/loss_metrics.py

@@ -16,6 +16,7 @@
 import torch.distributed as dist
 from torch.utils.tensorboard import SummaryWriter
 
+logging.basicConfig(level=logging.INFO)
 logger: logging.Logger = logging.getLogger(__name__)
 
 # create default device