MCPcopy Create free account
hub / github.com/Sin3DM/Sin3DM / __init__

Method __init__

src/diffusion/logger.py:155–169  ·  view source on GitHub ↗
(self, dir)

Source from the content-addressed store, hash-verified

153 """
154
155 def __init__(self, dir):
156 os.makedirs(dir, exist_ok=True)
157 self.dir = dir
158 self.step = 1
159 prefix = "events"
160 path = osp.join(osp.abspath(dir), prefix)
161 import tensorflow as tf
162 from tensorflow.python import pywrap_tensorflow
163 from tensorflow.core.util import event_pb2
164 from tensorflow.python.util import compat
165
166 self.tf = tf
167 self.event_pb2 = event_pb2
168 self.pywrap_tensorflow = pywrap_tensorflow
169 self.writer = pywrap_tensorflow.EventsWriter(compat.as_bytes(path))
170
171 def writekvs(self, kvs):
172 def summary_val(k, v):

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected