mirror of
https://github.com/RosettaCommons/foundry.git
synced 2026-06-04 13:24:22 +08:00
5e7b739ed3
* Add initial files * add files * Move projects.aa_design -> rfd3 * Make format * Delete test files * Add configs * Mc * Fixed tests * remove test files
54 lines
1.1 KiB
YAML
54 lines
1.1 KiB
YAML
# @package _global_
|
|
|
|
defaults:
|
|
# - /debug/default
|
|
# - override /logger: csv # turn off wandb
|
|
# - override /datasets: indexed_conditioned
|
|
# - override /trainer: aa_design_with_sequence # This has been absorbed into the default aa_design config
|
|
- override /model: unet-recycle
|
|
- _self_
|
|
|
|
name: main
|
|
tags:
|
|
- atom14
|
|
- unet
|
|
- uncond
|
|
- ${datasets.global_transform_args.central_atom}
|
|
|
|
trainer:
|
|
prevalidate: False
|
|
skip_optimizer_loading: False
|
|
|
|
loss:
|
|
verbose_diffusion_loss:
|
|
alpha_ca_atom: 1.0
|
|
sequence_head_loss:
|
|
weight: 0.5
|
|
max_t: 2
|
|
|
|
ckpt_path: null
|
|
|
|
# validation settings:
|
|
callbacks:
|
|
dump_validation_structures_callback:
|
|
dump_predictions: True
|
|
one_model_per_file: True
|
|
dump_trajectories: False
|
|
dump_every_n: 1
|
|
|
|
datasets:
|
|
diffusion_batch_size_inference: 10
|
|
|
|
model:
|
|
net:
|
|
diffusion_module:
|
|
use_sequence_head: True
|
|
# c_token: 384
|
|
# diffusion_transformer:
|
|
# n_block: 12
|
|
# inference_sampler:
|
|
# num_timesteps: 200
|
|
# noise_scale: 0.75
|
|
# step_scale: 1.5
|
|
# num_timesteps: 100
|