ViperEkura
/
AstrAI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refactor(khaosz/trainer): 移除未使用的导入模块

This commit is contained in:
ViperEkura 2025-10-04 21:45:53 +08:00
parent 9d5aa952e0
commit fc98d9b7e6
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
khaosz/trainer/strategy.py
View File

@ -5,11 +5,9 @@ import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
from torch import Tensor from torch import Tensor
from torch.optim import Optimizer from typing import Any, Literal, Tuple, Callable, Dict
from torch.utils.data import Dataset
from typing import Any, Literal, Optional, Tuple, Callable, Dict
from abc import ABC, abstractmethod 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): def get_logprobs(model:nn.Module, input_ids: Tensor, mask: Tensor, pad_token_id: int):