mirror of
https://github.com/FoundationVision/GLEE.git
synced 2025-06-03 14:51:17 +08:00
19 lines
634 B
Python
19 lines
634 B
Python
# Common training-related configs that are designed for "tools/lazyconfig_train_net.py"
|
|
# You can use your own instead, together with your own train_net.py
|
|
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=False,
|
|
find_unused_parameters=False,
|
|
fp16_compression=False,
|
|
),
|
|
checkpointer=dict(period=5000, max_to_keep=100), # options for PeriodicCheckpointer
|
|
eval_period=5000,
|
|
log_period=20,
|
|
device="cuda"
|
|
# ...
|
|
)
|