Jeney's picture
Upload folder using huggingface_hub
6a89c74
raw
history blame contribute delete
475 Bytes
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"
# ...
)