您最多选择25个主题
主题必须以中文或者字母或数字开头,可以包含连字符 (-),并且长度不得超过35个字符
33 行
749 B
33 行
749 B
estimator: UR3_single_cube_model
|
|
train:
|
|
dataset_zip_file_name_training: UR3_single_cube_training
|
|
batch_training_size: 20
|
|
accumulation_steps: 10
|
|
epochs: 120
|
|
beta_loss: 1
|
|
sample_size_train: 0
|
|
val:
|
|
dataset_zip_file_name_validation: UR3_single_cube_validation
|
|
batch_validation_size: 30
|
|
eval_freq: 4
|
|
sample_size_val: 0
|
|
test:
|
|
dataset_zip_file_name_test: UR3_single_cube_validation
|
|
batch_test_size: 30
|
|
sample_size_test: 0
|
|
dataset:
|
|
image_scale: 224
|
|
download_data_gcp: False
|
|
gcs_bucket: None
|
|
pose_estimation_gcs_path: None
|
|
symmetric: False
|
|
adam_optimizer:
|
|
lr: 0.0001
|
|
beta_1: 0.9
|
|
beta_2: 0.999
|
|
checkpoint:
|
|
load_dir_checkpoint: None
|
|
save_frequency: 1
|
|
system:
|
|
log_dir_system: /save/single_cube
|
|
data_root: /data
|