Update torch import
This commit is contained in:
		| @@ -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__) | ||||
|   | ||||
		Reference in New Issue
	
	Block a user