Open zhengshi119 opened 1 year ago
when i train biwi dataset，I the problem as follows:
Loading data... 0%| | 4/2187 [00:01<15:08, 2.40it/s] Traceback (most recent call last): File "main.py", line 151, in main() File "main.py", line 140, in main dataset = get_dataloaders(args) File "/workspace/FaceFormer-main/data_loader.py", line 98, in get_dataloaders train_data, valid_data, test_data, subjects_dict = read_data(args) File "/workspace/FaceFormer-main/data_loader.py", line 63, in read_data temp = templates[subject_id] KeyError: 'M5_e26'
Could anybody help me? Thanks!
when i train biwi dataset，I the problem as follows:
Loading data... 0%| | 4/2187 [00:01<15:08, 2.40it/s] Traceback (most recent call last): File "main.py", line 151, in
main()
File "main.py", line 140, in main
dataset = get_dataloaders(args)
File "/workspace/FaceFormer-main/data_loader.py", line 98, in get_dataloaders
train_data, valid_data, test_data, subjects_dict = read_data(args)
File "/workspace/FaceFormer-main/data_loader.py", line 63, in read_data
temp = templates[subject_id]
KeyError: 'M5_e26'
Could anybody help me? Thanks!