arsh121 commited on
Commit
dd606cb
·
1 Parent(s): 6041f57

Add model checkpoint and config

Browse files
checkpoints/timit_2/config.yaml ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ accumulate_grad_batches: 1
2
+ arch: 8 8 8 8 8 8 8 8
3
+ audio_num_mel_bins: 80
4
+ audio_sample_rate: 16000
5
+ base_config: configs/lipgen/lip_base.yaml
6
+ check_val_every_n_epoch: 10
7
+ clip_grad_norm: 1
8
+ cut_test_set: 10000
9
+ data_dir: data/timit
10
+ debug: false
11
+ dec_ffn_kernel_size: 9
12
+ dec_layers: 4
13
+ dropout: 0.2
14
+ dur_predictor_kernel: 3
15
+ enc_ffn_kernel_size: 9
16
+ enc_layers: 4
17
+ endless_ds: true
18
+ ffn_padding: SAME
19
+ frame_dir: data/timit/volunteers
20
+ gen_dir_name: ''
21
+ hidden_size: 256
22
+ img_channel: 3
23
+ img_h: 80
24
+ img_w: 160
25
+ imgdecoder_2D: true
26
+ infer: false
27
+ log_interval: 100
28
+ lr: 0.5
29
+ max_epochs: 1000
30
+ max_eval_sentences: 4
31
+ max_eval_tokens: 450
32
+ max_frames: 1550
33
+ max_input_tokens: 1550
34
+ max_sentences: -1
35
+ max_tokens: 450
36
+ max_updates: 110000
37
+ num_sanity_val_steps: 10
38
+ num_spk: 1
39
+ optimizer_adam_beta1: 0.9
40
+ optimizer_adam_beta2: 0.98
41
+ ph_dur_dir: data/timit/volunteers
42
+ predictor_hidden: 256
43
+ predictor_sg: true
44
+ profile_infer: false
45
+ raw_data_dir: ''
46
+ ref_hidden_stride_kernel:
47
+ - 256,3,9
48
+ - 256,3,9
49
+ - 256,2,9
50
+ - 256,2,9
51
+ - 256,2,9
52
+ ref_norm_layer: none
53
+ save_ckpt: true
54
+ seed: 1234
55
+ sep_dur_loss: true
56
+ sort_by_len: true
57
+ test_set_name: test
58
+ train_set_name: train
59
+ use_gt_dur: false
60
+ use_spk_id: false
61
+ use_training: false
62
+ val_check_interval: 2000
63
+ valid_set_name: valid
64
+ validate: false
65
+ vid_fps: 29.97
66
+ vid_loss: l1
67
+ vid_use_gt_dur: false
68
+ warmup_updates: 2000
69
+ weight_decay: 0
70
+ work_dir: checkpoints/timit_2
checkpoints/timit_2/model_ckpt_steps_32000.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9cbbe524116a1fa64414f13d19f229e85a1c66ab72d060826b79396ae7dc7d93
3
+ size 57131008