We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3e01030 commit b2082dbCopy full SHA for b2082db
hpctestlib/ml/pytorch/horovod.py
@@ -65,12 +65,14 @@ class pytorch_cnn_check(rfm.RunOnlyRegressionTest, pin_prefix=True):
65
tags = {'ml', 'cnn', 'horovod'}
66
67
@run_after('init')
68
- def prepare_test(self):
+ def update_descr(self):
69
# Get the python script
70
- script = self.executable.split()[1]
71
-
72
self.descr = (f'Distributed CNN training with PyTorch and Horovod '
73
f'(model: {self.model})')
+
+ @run_before('run')
74
+ def prepare_run(self):
75
+ script = self.executable.split()[1]
76
self.prerun_cmds = [
77
f'curl -LJO https://raw.githubusercontent.com/horovod/horovod/{self.benchmark_version}/examples/pytorch/{script}', # noqa: E501
78
]
0 commit comments