Update torch import

This commit is contained in:
Matthias 2023-04-11 19:38:24 +02:00
parent 4ab047dfa7
commit 9857675a5e

View File

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