dataset_mono: log when using heightmap

This commit is contained in:
Starbeamrainbowlabs 2022-12-06 19:30:11 +00:00
parent 468c150570
commit d997157f55
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -81,6 +81,7 @@ def make_dataset(filepaths, compression_type="GZIP", parallel_reads_multiplier=1
heightmap = None heightmap = None
if filepath_heightmap is not None: if filepath_heightmap is not None:
logger.info(f"Using heightmap from '{filepath_heightmap}'.")
heightmap = parse_heightmap(filepath_heightmap) heightmap = parse_heightmap(filepath_heightmap)
dataset = tf.data.TFRecordDataset(filepaths, dataset = tf.data.TFRecordDataset(filepaths,