lightning-tea

import torch
import torch.nn as nn
import torch.optim as optim
from torch.utils.data import DataLoader
import torchvision
from torchvision import transforms
from torchvision.models import efficientnet_b0, EfficientNet_B0_Weights

# 环境设置：检查GPU可用性
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

lightning-tea

import torch
import torch.nn as nn
import torch.optim as optim
from torch.utils.data import DataLoader
import torchvision
from torchvision import transforms
from torchvision.models import efficientnet_b0, EfficientNet_B0_Weights

# 环境设置
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

lightning-tea

import torch
import torch.nn as nn
import torch.optim as optim
from torch.utils.data import DataLoader
import torchvision
from torchvision import transforms
from torchvision.models import efficientnet_b0, EfficientNet_B0_Weights

# 环境设置
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")