Skip to content
Snippets Groups Projects
Commit 7adb61a4 authored by gong1's avatar gong1
Browse files

remove files for hparams

parent bbf348d6
Branches
Tags
No related merge requests found
Pipeline #105467 failed
{
"batch_size": 4,
"lr": 0.001,
"max_epochs":20,
"context_frames":12,
"loss_fun":"rmse",
"shuffle_on_val":false,
"recon_weight":0.6
}
{
"batch_size": 10,
"lr": 0.001,
"max_epochs": 2,
"context_frames": 12
}
{
"batch_size": 16,
"lr": 0.0002,
"beta1": 0.5,
"beta2": 0.999,
"l1_weight": 100.0,
"l2_weight": 0.0,
"kl_weight": 0.0,
"video_sn_vae_gan_weight": 0.0,
"video_sn_gan_weight": 0.1,
"vae_gan_feature_cdist_weight": 0.0,
"gan_feature_cdist_weight": 10.0,
"state_weight": 0.0,
"nz": 32,
"max_epochs":2,
"context_frames":12
}
{
"batch_size": 32,
"lr": 0.001,
"beta1": 0.9,
"beta2": 0.999,
"l1_weight": 1.0,
"l2_weight": 0.0,
"kl_weight": 1e-05,
"video_sn_vae_gan_weight": 0.0,
"video_sn_gan_weight": 0.0,
"state_weight": 0.0,
"nz": 32,
"max_epochs":2,
"context_frames":12
}
{
"batch_size": 10,
"lr": 0.001,
"nz":16,
"max_epochs":2,
"context_frames":12,
"weight_recon":1,
"loss_fun": "rmse",
"shuffle_on_val": true
}
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment