yodj/tf_official repository overview

version: '2.3'
services:
  lens_tube_train:
    image: "yodj/tf_official:2.2.1"
    runtime: nvidia
    logging:
      driver: 'json-file'
      options:
        max-size: '1G'
    environment:
      - NVIDIA_VISIBLE_DEVICES=0,1,2,3
    volumes:
      - ${configs_path}:${configs_path_in_docker}
      - ${model_dir_path}:${model_dir_in_docker}
      - ${datasets_path}:${dataset_path_in_docker}
    command: python ./vision/image_classification/classifier_trainer.py --mode=train_and_eval --model_type=resnet --dataset=imagenet --model_dir=${model_dir_in_docker} --data_dir=${dataset_path_in_docker} --config_file=${configs_path_in_docker} --params_override='runtime.num_gpus=4'