faster_rcnn_fbnetv3a_C4_FSDP.yaml

_BASE_: "faster_rcnn_fbnetv3a_C4.yaml"
D2GO_DATA:
  TEST:
    MAX_IMAGES: 50
MODEL:
  MODELING_HOOKS: ["FSDPModelingHook"]
DATASETS:
  TRAIN: ("coco_2017_val",)
  TEST: ("coco_2017_val",)
DATALOADER:
  NUM_WORKERS: 0
FSDP:
  ALGORITHM: "full"
  STATE_DICT_TYPE: "SHARDED_STATE_DICT"
  # AUTO_WRAP_POLICY: ""
  STATE_DICT_CPU_OFFLOAD: False
  STATE_DICT_RANK0_ONLY: True
LOAD_CKPT_TO_GPU: True
SOLVER:
  IMS_PER_BATCH: 32
  MAX_ITER: 5
  CHECKPOINT_PERIOD: 2
  AMP:
    ENABLED: False
    PRECISION: float16
TEST:
  EVAL_PERIOD: 10000
OUTPUT_DIR: /tmp/output