Отладка в Google Colab

Question

Я запускаю следующий фрагмент кода в Google Colab в одной ячейке:

%debug
# Create tensors of shape (10, 3) and (10, 2).
x = torch.randn(10, 3)
y = torch.randn(10, 2)

# Build a fully connected layer.
linear = nn.Linear(3, 2)
print ('w: ', linear.weight)
print ('b: ', linear.bias)

Я хочу отладить фрагмент кода (шаг за шагом, чтобы понять, что происходит).Я хочу шагнуть внутрь функции nn.Linear.

Однако, когда я прохожу, он вообще не входит в функцию.Есть ли способ пройти через nn.Linear строка за строкой?Кроме того, как именно я могу установить точку останова в nn.Linear?Кроме того, я хочу шаг за шагом пересечь фрагмент кода.Однако, как показано на рисунке, команда step автоматически выполняет и выполняет оператор print.

Answer 1

Начиная с Python 3.7 вы можете использовать встроенную функцию точки останова .Если это недоступно, вы можете использовать import pdb; pdb.set_trace() вместо.

Если вы хотите выполнить следующую строку, вы можете попробовать n (next) вместо s (step).