Open wr0124 opened 1 week ago
python3 -W ignore::UserWarning train.py \ --dataroot /path/to/dataset \ --checkpoints_dir /path/to/checkpoints \ --name test_debug \ --gpu_ids 3 \ --data_relative_paths \ --model_type palette \ --data_dataset_mode self_supervised_vid_mask_online \ --train_batch_size 2 \ --train_iter_size 1 \ --test_batch_size 1 \ --data_num_threads 1 \ --model_input_nc 3 \ --model_output_nc 3 \ --data_relative_paths \ --train_G_ema \ --train_optim adamw \ --G_netG unet_vid \ --data_online_creation_rand_mask_A \ --train_G_lr 0.0002 \ --dataaug_no_rotate \ --G_diff_n_timestep_train 7 \ --G_diff_n_timestep_test 3 \ --data_temporal_frame_step 1 \ --alg_diffusion_cond_image_creation computed_sketch \ --alg_diffusion_cond_computed_sketch_list canny \ --alg_diffusion_cond_sketch_canny_range 0 190 \ --alg_diffusion_vid_canny_dropout 0.3 0.7 \ --train_compute_metrics_test \ --train_metrics_every 2 \ --train_metrics_list PSNR LPIPS SSIM \ --output_print_freq 1 \ --output_display_freq 1 \ --data_temporal_number_frames 8 \ --data_online_creation_crop_size_A 64 \ --data_online_creation_crop_size_B 32 \ --data_crop_size 32 \ --data_load_size 32 \ --data_online_creation_mask_square_A \ --with_amp \ --with_tf32 \ ~
python3 -W ignore::UserWarning train.py \ --dataroot /path/to/dataset \ --checkpoints_dir /path/to/checkpoints \ --name test_debug \ --gpu_ids 3 \ --data_relative_paths \ --model_type palette \ --data_dataset_mode self_supervised_vid_mask_online \ --train_batch_size 2 \ --train_iter_size 1 \ --test_batch_size 1 \ --data_num_threads 1 \ --model_input_nc 3 \ --model_output_nc 3 \ --data_relative_paths \ --train_G_ema \ --train_optim adamw \ --G_netG unet_vid \ --data_online_creation_rand_mask_A \ --train_G_lr 0.0002 \ --dataaug_no_rotate \ --G_diff_n_timestep_train 7 \ --G_diff_n_timestep_test 3 \ --data_temporal_frame_step 1 \ --alg_diffusion_cond_image_creation computed_sketch \ --alg_diffusion_cond_computed_sketch_list canny \ --alg_diffusion_cond_sketch_canny_range 0 190 \ --alg_diffusion_vid_canny_dropout 0.3 0.7 \ --train_compute_metrics_test \ --train_metrics_every 2 \ --train_metrics_list PSNR LPIPS SSIM \ --output_print_freq 1 \ --output_display_freq 1 \ --data_temporal_number_frames 8 \ --data_online_creation_crop_size_A 64 \ --data_online_creation_crop_size_B 32 \ --data_crop_size 32 \ --data_load_size 32 \ --data_online_creation_mask_square_A \ --with_amp \ --with_tf32 \ ~