Add TODO.

Author: Xreki

graph_net/torch/utils.py

@@ -272,6 +272,7 @@ def replay_tensor(info):
     if dtype is torch.bool:
         return (torch.randn(size=shape) > 0.5).to(dtype).to(device)
     tensor = torch.randn(size=shape).to(dtype).to(device) * std * 0.2 + mean
+    # TODO(Xreki): remove this ugly code, and change the weight_meta instead.
     if name.startswith("L_self_modules") and "buffers_running_var" in name:
         tensor = torch.clip(tensor, min=0)
     return tensor