Update torch import

This commit is contained in:
Matthias 2023-04-11 19:38:24 +02:00
parent 4ab047dfa7
commit 9857675a5e
1 changed files with 1 additions and 1 deletions

View File

@ -2,7 +2,7 @@ import logging
from typing import List
import torch
import torch.nn as nn
from torch import nn
logger = logging.getLogger(__name__)