deepdrivemd.models.aae_stream.config
Classes
|
- class deepdrivemd.models.aae_stream.config.Point3dAAEConfig(*args: Any, **kwargs: Any)
- adios_xml_agg
- adios_xml_agg_4ml
- agg_dir = PosixPath('.')
- batch_size = 32
- checkpoint_dir
- cms_transform = True
- decoder_affine_widths = [64, 128, 512, 1024]
- decoder_bias = True
- decoder_relu_slope = 0.0
- discriminator_affine_widths = [512, 128, 64]
- discriminator_bias = True
- discriminator_relu_slope = 0.0
- encoder_bias = True
- encoder_filters = [64, 128, 256, 256, 512]
- encoder_kernels = [5, 5, 3, 1, 1]
- encoder_relu_slope = 0.0
- epochs = 30
- experiment_directory
- init_weights = ''
- init_weights_path
- input_path = PosixPath('/p/gpfs1/yakushin/Outputs/305t/molecular_dynamics_runs/stage0000/task0000/0/trajectory.bp')
- lambda_gp = 10.0
- lambda_rec = 0.5
- latent_dim = 16
- max_loss = 10000
- max_steps = 8000
- min_step_increment = 5000
- model = 'aae'
- model_tag = 'aae'
- node_local_path = PosixPath('/tmp')
- noise_mu = 0.0
- noise_std = 1.0
- num_agg = 12
- num_data_workers = 16
- num_features = 0
- num_points = 200
- output_path = PosixPath('TODO')
- prefetch_factor = 2
- published_model_dir
- read_batch = 10000
- reinit = False
- resume_checkpoint = None
- scalar_dset_names = []
- scalar_requires_grad = False
- seed = 333
- shuffle = True
- split_pct = 0.8
- stage_idx = 0
- task_idx = 0
- timeout1 = 30
- timeout2 = 10
- use_model_checkpoint = True