pub_soft/foldingdiff
3
0
Fork 0
mirror of https://github.com/microsoft/foldingdiff.git synced 2026-06-04 13:30:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Typo bugfix

Browse Source
This commit is contained in:
Kevin Wu
2022-08-26 21:09:19 +00:00
parent 68b7b4d25a
commit 435eedbd22
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
protdiff/modelling.py
View File

@@ -226,7 +226,7 @@ class BertForDiffusion(BertPreTrainedModel, pl.LightningModule):
time_encoding: Literal["gaussian_fourier", "sinusoidal"] = "sinusoidal",
decoder: Literal["linear", "mlp"] = "linear",
lr: float = 1e-4,
loss: Union[Callable, Literal["l1", "l1_smooth"]] = "l1_smooth",
loss: Union[Callable, Literal["l1", "smooth_l1"]] = "smooth_l1",
l2: float = 0.0,
l1: float = 0.0,
circle_reg: float = 0.0,