libokj commited on
Commit
d35cf1f
·
1 Parent(s): 57ed866

Update configs/preset/m_graph_dta.yaml

Browse files
Files changed (1) hide show
  1. configs/preset/m_graph_dta.yaml +1 -4
configs/preset/m_graph_dta.yaml CHANGED
@@ -16,7 +16,4 @@ data:
16
  atom_features:
17
  _target_: deepscreen.models.predictors.m_graph_dta.atom_features
18
  _partial_: true
19
- batch_size: 512
20
-
21
- trainer:
22
- precision: 'bf16'
 
16
  atom_features:
17
  _target_: deepscreen.models.predictors.m_graph_dta.atom_features
18
  _partial_: true
19
+ batch_size: 512