refactor(khaosz/trainer): 移除未使用的导入模块
This commit is contained in:
parent
9d5aa952e0
commit
fc98d9b7e6
|
|
@ -5,11 +5,9 @@ import torch.nn as nn
|
|||
import torch.nn.functional as F
|
||||
|
||||
from torch import Tensor
|
||||
from torch.optim import Optimizer
|
||||
from torch.utils.data import Dataset
|
||||
from typing import Any, Literal, Optional, Tuple, Callable, Dict
|
||||
from typing import Any, Literal, Tuple, Callable, Dict
|
||||
from abc import ABC, abstractmethod
|
||||
from dataclasses import asdict, dataclass, field
|
||||
from dataclasses import dataclass, field
|
||||
|
||||
|
||||
def get_logprobs(model:nn.Module, input_ids: Tensor, mask: Tensor, pad_token_id: int):
|
||||
|
|
|
|||
Loading…
Reference in New Issue