Spaces:
Running
Running
File size: 523 Bytes
684943d |
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 |
# validation
val_steps=10
wandb=False
record_time = True
# autoencoder config
# exp_name = "AE-gpu8-bs1-channel-16"
# autoencoder_name = "ae-channel-16"
exp_name = "test"
autoencoder_name = "ae-channel-3"
dataset_name = "lvis"
lseg_weights="model_zoo/lseg/demo_e200.ckpt"
# loss weights
perceptual_loss_weight = 0.1 # use vgg is not None and more than 0
kl_loss_weight = 1e-6
mixed_strategy = "mixed_video_image"
mixed_image_ratio = 0.2
use_real_rec_loss = False
use_z_rec_loss = True
use_image_identity_loss = True |