Skip to content

Commit 2586ee3

Browse files
committed
Merge: [SE3Transformer/DGLPyT] Move results to workdir
2 parents ef5241e + a23be21 commit 2586ee3

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

DGLPyTorch/DrugDiscovery/SE3Transformer/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -285,7 +285,7 @@ To train your model using mixed or TF32 precision with Tensor Cores or FP32, per
285285
3. Start an interactive session in the NGC container to run training/inference.
286286
```
287287
mkdir -p results
288-
docker run -it --runtime=nvidia --shm-size=8g --ulimit memlock=-1 --ulimit stack=67108864 --rm -v ${PWD}/results:/results se3-transformer:latest
288+
docker run -it --runtime=nvidia --shm-size=8g --ulimit memlock=-1 --ulimit stack=67108864 --rm -v ${PWD}/results:/workspace/se3-transformer/results se3-transformer:latest
289289
```
290290
291291
4. Start training.

DGLPyTorch/DrugDiscovery/SE3Transformer/se3_transformer/runtime/arguments.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@
3333
paths = PARSER.add_argument_group('Paths')
3434
paths.add_argument('--data_dir', type=pathlib.Path, default=pathlib.Path('./data'),
3535
help='Directory where the data is located or should be downloaded')
36-
paths.add_argument('--log_dir', type=pathlib.Path, default=pathlib.Path('/results'),
36+
paths.add_argument('--log_dir', type=pathlib.Path, default=pathlib.Path('./results'),
3737
help='Directory where the results logs should be saved')
3838
paths.add_argument('--dllogger_name', type=str, default='dllogger_results.json',
3939
help='Name for the resulting DLLogger JSON file')

0 commit comments

Comments
 (0)