Skip to content

Commit 7a3ee90

Browse files
committed
implement workaround for cuda toolkit build difference of torch and torchvision in CI that was resulting in a segfault triggered by sentencepiece upon pytorch_lightning import
1 parent f6ec983 commit 7a3ee90

File tree

2 files changed

+4
-13
lines changed

2 files changed

+4
-13
lines changed

lightning_examples/finetuning-scheduler/.meta.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ author: "[Dan Dale](https://github.com/speediedan)"
33
created: 2021-11-29
44
updated: 2022-03-29
55
license: CC BY-SA
6-
build: 1
6+
build: 2
77
tags:
88
- finetuning
99
description: |

lightning_examples/finetuning-scheduler/finetuning-scheduler.py

Lines changed: 3 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,3 @@
1-
# ---
2-
# jupyter:
3-
# jupytext:
4-
# cell_metadata_filter: -all
5-
# formats: ipynb,py:percent
6-
# text_representation:
7-
# extension: .py
8-
# format_name: percent
9-
# format_version: '1.3'
10-
# jupytext_version: 1.13.8
11-
# ---
12-
131
# %% [markdown]
142
# ## Scheduled Finetuning
153
#
@@ -176,6 +164,9 @@
176164

177165
import datasets
178166
import pytorch_lightning as pl
167+
168+
# temporary workaround until implementation of https://github.com/PyTorchLightning/lightning-tutorials/pull/155
169+
import sentencepiece as sp # noqa:F401
179170
import torch
180171
from pytorch_lightning.callbacks import EarlyStopping, ModelCheckpoint
181172
from pytorch_lightning.loggers.tensorboard import TensorBoardLogger

0 commit comments

Comments
 (0)