We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b892a3a commit 4105798Copy full SHA for 4105798
2 files changed
README.md
@@ -57,9 +57,9 @@ runai project set <your project>
57
```
58
59
60
-runai workspace submit openpulse-training \
+runai workspace submit openpulse-training2 \
61
-i ghcr.io/sdsc-ordes/open-pulse-graph-classifier-train:latest \
62
- --gpu 1 \
+ --gpu-request-type portion --gpu-portion-request 0.9 \
63
--image-pull-policy Always \
64
--attach
65
src/classifier/processing/loaders.py
@@ -8,7 +8,7 @@ def build_loader(
8
split_idx,
9
shuffle,
10
batch_size=32,
11
- num_neighbors=[10, 5]
+ num_neighbors=[10, 5],
12
random_state=42,
13
):
14
return NeighborLoader(
0 commit comments