Skip to content

Commit

Permalink
input_full.yaml correct
Browse files Browse the repository at this point in the history
  • Loading branch information
YutackPark authored and YutackPark committed Aug 28, 2023
1 parent d00854f commit 7d87396
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions example_inputs/training/input_full.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -61,10 +61,11 @@ train:
# Continue training from a checkpoint. If you augment the dataset, recalculated shift, scale, avg_num_neigh will be used
# Also, you can enable train_shift_scale or train_avg_num_neigh
# input.yaml for original and continued training should have the same model hyperparameters except for avg_num_neigh and shift scale
# Note that it will overwrite log and .pt or .pth files if the training continue on the same directory.
# continue:
# reset_optimizer: False
# reset_scheduler: False
# checkpoint: False
# checkpoint: 'path_to_previous_checkpoint/checkpoint_best.pth'

data:
batch_size: 2 # Batch size. If training fails due to memory shortage, lower this value
Expand All @@ -77,6 +78,3 @@ data:

save_dataset_path: 'total' # Save the preprocessed dataset for later use without the structure_list
save_by_label: False # Save the dataset by labels specified in the structure_list

save_dataset_path: 'total' # Save preprocessed dataset for later use with out structure_list.
save_by_label: False # Save dataset by labels specified on structure_list

0 comments on commit 7d87396

Please sign in to comment.