@@ -14,6 +14,8 @@ services:
environment:
COMPILE: ${COMPILE:-0}
PYTORCH_CUDA_ALLOC_CONF: ${PYTORCH_CUDA_ALLOC_CONF:-expandable_segments:True}
+ TORCH_LOGS: +dynamo
+ TORCHDYNAMO_VERBOSE: 1
# GPU (remove this block if CPU-only):
deploy:
resources: