epos-ai/platform-demo-scripts
3
0
Fork 0
Code Issues Pull Requests Activity

restored default Phasenet phases labels

Browse Source
This commit is contained in:
Krystyna Milian 2024-02-26 23:39:10 +01:00
parent 4658b8d866
commit 87de2e7a6c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/models.py
View File

@ -131,7 +131,7 @@ class PhaseNetLit(SeisBenchModuleLit):
self.sigma = sigma self.sigma = sigma
self.sample_boundaries = sample_boundaries self.sample_boundaries = sample_boundaries
self.loss = vector_cross_entropy self.loss = vector_cross_entropy
self.model = sbm.PhaseNet(phases="PN", **kwargs) self.model = sbm.PhaseNet(**kwargs)
def forward(self, x): def forward(self, x):
return self.model(x) return self.model(x)