forgot to commit this one, just logging
This commit is contained in:
Родитель
8452726567
Коммит
74e14f731d
|
@ -272,12 +272,15 @@ def main():
|
||||||
|
|
||||||
logger.info('Creating data sources ...')
|
logger.info('Creating data sources ...')
|
||||||
if _FAKE:
|
if _FAKE:
|
||||||
|
logger.info("Using {} images of fake data".format(_DATA_LENGTH))
|
||||||
train_source = FakeDataSource(total_n_images=_DATA_LENGTH,
|
train_source = FakeDataSource(total_n_images=_DATA_LENGTH,
|
||||||
dim=(_HEIGHT, _WIDTH),
|
dim=(_HEIGHT, _WIDTH),
|
||||||
channels=_CHANNELS,
|
channels=_CHANNELS,
|
||||||
n_classes=_NUMCLASSES)
|
n_classes=_NUMCLASSES)
|
||||||
test_source = None
|
test_source = None
|
||||||
else:
|
else:
|
||||||
|
logging.info(
|
||||||
|
"Using ImageNet dataset with {} images".format(_DATA_LENGTH))
|
||||||
data_path = os.getenv('AZ_BATCHAI_INPUT_TRAIN')
|
data_path = os.getenv('AZ_BATCHAI_INPUT_TRAIN')
|
||||||
logger.info("model_path: {}".format(model_path))
|
logger.info("model_path: {}".format(model_path))
|
||||||
logger.info("data_path: {}".format(data_path))
|
logger.info("data_path: {}".format(data_path))
|
||||||
|
|
Загрузка…
Ссылка в новой задаче