File size: 761 Bytes
59630ba | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 | defaults:
- base_pytorch_algo
model:
hidden_size: 128
temporal_length: 17
z_channels: 8
hidden_size_mult: [1, 2, 4, 4]
attn_resolutions: []
dropout: 0.0
resolution: ${dataset.resolution}
embed_dim: ${algorithm.model.z_channels}
num_res_blocks: 2
is_causal: True
first_padding_mode: same
loss:
disc_start: 60001
disc_weight: 0.5
kl_weight: 5e-6
perceptual_weight: 1.0
loss_type: l1
training:
warmup_steps: 20000
gradient_clip_val: 1.0
optimizer_beta: [0.9, 0.99]
video_lengths: ["${algorithm.model.temporal_length}"]
video_length_probs: [1.0]
validation:
video_lengths: ["${algorithm.model.temporal_length}"]
logging:
metrics: [lpips, fid, fvd, is, mse, ssim, psnr]
loss_freq: 100
max_num_videos: 64
|