Update torch import
This commit is contained in:
parent
4ab047dfa7
commit
9857675a5e
@ -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__)
|
||||
|
Loading…
Reference in New Issue
Block a user