Skip to content
Snippets Groups Projects
Commit 5a3fd8fb authored by febert's avatar febert
Browse files

Revert "added setup.py"

This reverts commit 2c6c6954.
parent 2c6c6954
Branches
Tags
No related merge requests found
Showing
with 0 additions and 246 deletions
......@@ -99,10 +99,3 @@ ENV/
# mypy
.mypy_cache/
# tensorflow model files:
model-*
events.out*
.idea/
model_checkpoint_path: "model-300000"
all_model_checkpoint_paths: "model-290000"
all_model_checkpoint_paths: "model-295000"
all_model_checkpoint_paths: "model-300000"
{
"context_frames": 2,
"crop_size": 0,
"force_time_shift": false,
"frame_skip": 0,
"scale_size": 0,
"sequence_length": 15,
"time_shift": 3,
"use_state": true
}
\ No newline at end of file
{
"ablation_conv_rnn_norm": false,
"acvideo_gan_weight": 0.0,
"acvideo_vae_gan_weight": 0.0,
"ae_l2_weight": 0.0,
"batch_size": 16,
"beta1": 0.9,
"beta2": 0.999,
"clip_length": 10,
"context_frames": 2,
"context_images_background": false,
"conv_rnn": "lstm",
"d_conditional": true,
"d_downsample_layer": "conv_pool2d",
"d_net": "legacy",
"d_use_gt_inputs": true,
"decay_steps": [
200000,
300000
],
"dependent_mask": true,
"dilation_rate": [
1,
1
],
"downsample_layer": "conv_pool2d",
"e_net": "n_layer",
"end_lr": 0.0,
"feature_l2_weight": 0.0,
"first_image_background": true,
"gan_feature_cdist_weight": 0.0,
"gan_feature_l2_weight": 0.0,
"gan_loss_type": "LSGAN",
"gan_weight": 0.0,
"generate_scratch_image": false,
"image_gan_weight": 0.0,
"image_sn_gan_weight": 0.0,
"image_sn_vae_gan_weight": 0.0,
"image_vae_gan_weight": 0.0,
"kernel_size": [
5,
5
],
"kl_anneal": "linear",
"kl_anneal_k": -1.0,
"kl_anneal_steps": [
50000,
100000
],
"kl_weight": 0.001,
"l1_weight": 1.0,
"l2_weight": 0.0,
"last_frames": 2,
"lr": 0.001,
"max_steps": 300000,
"n_layers": 3,
"ndf": 32,
"nef": 64,
"ngf": 32,
"norm_layer": "instance",
"num_samples": 8,
"num_transformed_images": 4,
"nz": 8,
"prev_image_background": true,
"repeat": 3,
"rnn": "lstm",
"schedule_sampling": "inverse_sigmoid",
"schedule_sampling_k": 900.0,
"schedule_sampling_steps": [
0,
100000
],
"sequence_length": 15,
"state_weight": 0.0001,
"transformation": "flow",
"tuple_gan_weight": 0.0,
"tuple_vae_gan_weight": 0.0,
"tv_weight": 0.0,
"upsample_layer": "upsample_conv2d",
"use_e_rnn": false,
"use_rnn_z": true,
"vae_gan_weight": 0.0,
"vgg_cdist_weight": 0.0,
"video_gan_weight": 0.0,
"video_sn_gan_weight": 0.0,
"video_sn_vae_gan_weight": 0.0,
"video_vae_gan_weight": 0.0,
"where_add": "all",
"z_l1_weight": 0.0
}
\ No newline at end of file
{
"checkpoint": null,
"dataset": "cartgripper",
"dataset_hparams": null,
"dataset_hparams_dict": null,
"eval_summary_freq": 0,
"gif_freq": 0,
"gpu_mem_frac": 0,
"image_summary_freq": 5000,
"input_dir": "data/cartgripper_updown_sact",
"logs_dir": "logs/cartgripper/vae",
"metrics_freq": 0,
"model": "savp",
"model_hparams": "transformation=flow,last_frames=2,generate_scratch_image=false,batch_size=16",
"model_hparams_dict": "hparams/bair/ours_vae_l1/model_hparams.json",
"output_dir": "logs/cartgripper/vae/model.savp.transformation.flow.last_frames.2.generate_scratch_image.false.batch_size.16",
"progress_freq": 100,
"resume": false,
"save_freq": 5000,
"seed": null,
"summary_freq": 1000,
"val_input_dirs": null
}
\ No newline at end of file
model_checkpoint_path: "model-300000"
{
"context_frames": 2,
"crop_size": 0,
"force_time_shift": false,
"frame_skip": 0,
"scale_size": 0,
"sequence_length": 12,
"time_shift": 2,
"use_state": false
}
\ No newline at end of file
{
"ablation_conv_rnn_norm": false,
"acvideo_gan_weight": 0.0,
"acvideo_vae_gan_weight": 0.0,
"ae_l2_weight": 0.0,
"batch_size": 16,
"beta1": 0.5,
"beta2": 0.999,
"clip_length": 10,
"context_frames": 2,
"context_images_background": false,
"conv_rnn": "lstm",
"d_conditional": true,
"d_downsample_layer": "conv_pool2d",
"d_net": "legacy",
"d_use_gt_inputs": true,
"decay_steps": [
200000,
300000
],
"dependent_mask": true,
"dilation_rate": [
1,
1
],
"downsample_layer": "conv_pool2d",
"e_net": "n_layer",
"end_lr": 0.0,
"feature_l2_weight": 0.0,
"first_image_background": true,
"gan_feature_cdist_weight": 0.0,
"gan_feature_l2_weight": 0.0,
"gan_loss_type": "GAN",
"gan_weight": 0.0,
"generate_scratch_image": true,
"image_gan_weight": 0.0,
"image_sn_gan_weight": 0.0,
"image_sn_vae_gan_weight": 0.0,
"image_vae_gan_weight": 0.0,
"kernel_size": [
5,
5
],
"kl_anneal": "linear",
"kl_anneal_k": -1.0,
"kl_anneal_steps": [
50000,
100000
],
"kl_weight": 0.1,
"l1_weight": 100.0,
"l2_weight": 0.0,
"last_frames": 1,
"lr": 0.0002,
"n_layers": 3,
"ndf": 32,
"nef": 64,
"ngf": 32,
"norm_layer": "instance",
"num_samples": 8,
"num_transformed_images": 4,
"nz": 8,
"prev_image_background": true,
"repeat": 2,
"rnn": "lstm",
"schedule_sampling": "inverse_sigmoid",
"schedule_sampling_k": 900.0,
"schedule_sampling_steps": [
0,
100000
],
"sequence_length": 12,
"state_weight": 0.0,
"transformation": "cdna",
"tuple_gan_weight": 0.0,
"tuple_vae_gan_weight": 0.0,
"tv_weight": 0.0,
"upsample_layer": "upsample_conv2d",
"use_e_rnn": false,
"use_rnn_z": true,
"vae_gan_weight": 0.0,
"vgg_cdist_weight": 0.0,
"video_gan_weight": 0.0,
"video_sn_gan_weight": 1.0,
"video_sn_vae_gan_weight": 1.0,
"video_vae_gan_weight": 0.0,
"where_add": "all",
"z_l1_weight": 0.0
}
\ No newline at end of file
{
"dataset": "softmotion",
"model": "savp"
}
\ No newline at end of file
# setup.py
from setuptools import setup
setup(
name='alex_video_prediction',
version='0.1.0',
packages=['video_prediction'],
)
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment