train = dict( output_dir="./output", init_checkpoint="", max_iter=90000, amp=dict(enabled=False), # options for Automatic Mixed Precision ddp=dict( # options for DistributedDataParallel broadcast_buffers=True, find_unused_parameters=False, fp16_compression=True, ), checkpointer=dict(period=5000, max_to_keep=100), # options for PeriodicCheckpointer eval_period=5000, log_period=20, device="cuda", seed=42 # ... )