Zhang Peiyuan
d417e4c7c4
typo add upload command add add ema_transform add ema transformer remove harcode ok ema remove hardcode update env update script; no sp revert to sp=4, sp bs=2, full shard add aws efo env distributed validation readme distributed validation add gupload linear range; fix uncond; multi guidance validation add script
23 lines
516 B
Python
23 lines
516 B
Python
|
|
import sys
|
|
import pdb
|
|
import os
|
|
|
|
def main_print(content):
|
|
if int(os.environ['LOCAL_RANK']) <= 0:
|
|
print(content)
|
|
|
|
#ForkedPdb().set_trace()
|
|
class ForkedPdb(pdb.Pdb):
|
|
"""A Pdb subclass that may be used
|
|
from a forked multiprocessing child
|
|
|
|
"""
|
|
def interaction(self, *args, **kwargs):
|
|
_stdin = sys.stdin
|
|
try:
|
|
sys.stdin = open('/dev/stdin')
|
|
pdb.Pdb.interaction(self, *args, **kwargs)
|
|
finally:
|
|
sys.stdin = _stdin
|
|
|