cifar10_multi_gpu_train_highway.py 文件源码

def main(argv=None):  # pylint: disable=unused-argument
  setConfig()
  config = network_config.getConfig()
  train_dir = config['train_dir']

  cifar10.maybe_download_and_extract()
  if gfile.Exists(train_dir):
    gfile.DeleteRecursively(train_dir)
  gfile.MakeDirs(train_dir)
  train()