Skip to content

Commit 3ea7543

Browse files
committed
Add information about pytorch-02 branch
1 parent 602507a commit 3ea7543

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

README.md

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,13 @@ To generate the flow weights, use `python i3d_tf_to_pt.py --flow`.
1414

1515
You can also generate both in one run by using both flags simultaneously `python i3d_tf_to_pt.py --rgb --flow`.
1616

17+
Note that the master version requires PyTorch compiled from master because it relies the recent addition of ConstantPad3d in the torch source code.
18+
19+
If you want to use pytorch 0.2 checkout the branch pytorch-02 which contains a simplified model with even padding on all sides (and the pytorch weight checkpoints).
20+
The difference is that the 'SAME' option for padding in tensorflow allows it to pad unevenly both sides of a dimension, an effect reproduced on the master branch.
21+
22+
This simpler model produces scores a bit closer to the original tensorflow model on the demo sample and is also a bit faster.
23+
1724
## Demo
1825

1926
There is a slight drift in the weights that impacts the predictions, however, it seems to only marginally affect the final predictions, and therefore, the converted weights should serve as a valid initialization for further finetuning.

0 commit comments

Comments
 (0)