Faaz commited on
Commit
1b432fd
·
1 Parent(s): 34d0e72

Force line-buffered stdout for real-time log output in docker

Browse files
Files changed (1) hide show
  1. src/training/mindi_trainer.py +5 -0
src/training/mindi_trainer.py CHANGED
@@ -21,11 +21,16 @@ from __future__ import annotations
21
 
22
  import json
23
  import math
 
24
  import time
25
  from dataclasses import dataclass, field
26
  from pathlib import Path
27
  from typing import Any, Iterator, Optional
28
 
 
 
 
 
29
  import torch
30
  import torch.nn as nn
31
  from PIL import Image
 
21
 
22
  import json
23
  import math
24
+ import sys
25
  import time
26
  from dataclasses import dataclass, field
27
  from pathlib import Path
28
  from typing import Any, Iterator, Optional
29
 
30
+ # Force unbuffered stdout for docker exec -d log visibility
31
+ if not sys.stdout.line_buffering:
32
+ sys.stdout.reconfigure(line_buffering=True)
33
+
34
  import torch
35
  import torch.nn as nn
36
  from PIL import Image