alibabasglab commited on
Commit
256f028
·
verified ·
1 Parent(s): 5d43c9a

Delete checkpoints/AV_MossFormer2_TSE_16K

Browse files
checkpoints/AV_MossFormer2_TSE_16K/config.yaml DELETED
@@ -1,55 +0,0 @@
1
- ## Config file
2
-
3
- # Log
4
- seed: 777
5
- use_cuda: 1 # 1 for True, 0 for False
6
-
7
- # dataset
8
- speaker_no: 2
9
- mix_lst_path: ./data/allData/voxceleb2/mixture_data_list_2mix_pretrain.csv
10
- audio_direc: /mnt/nas_sg/wulanchabu/zexu.pan/datasets/voxceleb2/audio_clean
11
- reference_direc: /mnt/nas_sg/wulanchabu/zexu.pan/datasets/ # not used
12
- audio_sr: 16000
13
- ref_sr: 25
14
-
15
- # dataloader
16
- num_workers: 4
17
- batch_size: 2 # 4-GPU training with a total effective batch size of 8
18
- accu_grad: 0
19
- effec_batch_size: 4 # per GPU, only used if accu_grad is set to 1, must be multiple times of batch size
20
- max_length: 5 # truncate the utterances in dataloader, in seconds
21
-
22
- # network settings
23
- init_from: checkpoints/log_2024-09-30(09:49:14) # 'None' or a log name 'log_2024-07-22(18:12:13)'
24
- causal: 0 # 1 for True, 0 for False
25
- network_reference:
26
- cue: lip # lip or speech or gesture or EEG
27
- backbone: resnet18 # resnet18 or shufflenetV2 or blazenet64
28
- emb_size: 256 # resnet18:256
29
- network_audio:
30
- backbone: mossformer2
31
- encoder_kernel_size: 16
32
- encoder_out_nchannels: 512
33
- encoder_in_nchannels: 1
34
-
35
- masknet_numspks: 1
36
- masknet_chunksize: 250
37
- masknet_numlayers: 1
38
- masknet_norm: "ln"
39
- masknet_useextralinearlayer: False
40
- masknet_extraskipconnection: True
41
-
42
- intra_numlayers: 24
43
- intra_nhead: 8
44
- intra_dffn: 1024
45
- intra_dropout: 0
46
- intra_use_positional: True
47
- intra_norm_before: True
48
-
49
-
50
- # optimizer
51
- loss_type: hybrid # "snr", "sisdr", "hybrid"
52
- init_learning_rate: 0.00015
53
- max_epoch: 150
54
- clip_grad_norm: 5
55
-
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
checkpoints/AV_MossFormer2_TSE_16K/last_best_checkpoint DELETED
@@ -1 +0,0 @@
1
- last_best_checkpoint.pt
 
 
checkpoints/AV_MossFormer2_TSE_16K/last_best_checkpoint.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:591a51cfea0a4038b006793308a42b74a661a58b5e0afac20f1def0700b44ae6
3
- size 734561014