Skip to content

Commit d88557c

Browse files
authored
Update README.md
1 parent d6cbf32 commit d88557c

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

examples/batfd/README.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,8 @@ Train the BATFD or BATFD+ model using a TOML configuration file (e.g., `batfd.to
2121
python train.py --config ./batfd.toml --data_root /path/to/AV-Deepfake1M-PlusPlus
2222
```
2323

24+
If you meet the NaN issue when training BA-TFD+, that might be caused by the bug in PyTorch self attention ops, upgrading or changing the PyTorch version can solve it.
25+
2426
### Output
2527

2628
* **Checkpoints:** Model checkpoints are saved under `./ckpt/xception/`. The last checkpoint is saved as `last.ckpt`.

0 commit comments

Comments
 (0)