Description of model keys doesn't match GitHub repo
#1
by ekiefl - opened
The description card says:
original_v_48_002 - Trained for 2 epochs
original_v_48_010 - Trained for 10 epochs
original_v_48_020 - Trained for 20 epochs (recommended)
original_v_48_030 - Trained for 30 epochs
But in the original protein MPNN repo it says:
Models provided for the vanilla MPNN were trained with default flags:
* `v_48_002.pt` - `--num_neighbors 48 --backbone_noise 0.02 --num_epochs 150`
* `v_48_010.pt` - `--num_neighbors 48 --backbone_noise 0.10 --num_epochs 150`
* `v_48_020.pt` - `--num_neighbors 48 --backbone_noise 0.20 --num_epochs 150`
Please verify and update accordingly.
Good catch, I'll update. Didn't realize I had even made these model ports public yet