beginner/hyperparameter_tuning_tutorial
Run in Google Colab
Colab
Download Notebook
Notebook
View on GitHub
GitHub
Note
Go to the end to download the full example code.
Hyperparameter tuning with Ray Tune#
Created On: Aug 31, 2020 | Last Updated: Jun 24, 2025 | Last Verified: Nov 05, 2024
Hyperparameter tuning can make the difference between an average model and a highly accurate one. Often simple things like choosing a different learning rate or changing a network layer size can have a dramatic impact on your model performance.
Fortunately, there are tools that help with finding the best combination of parameters. Ray Tune is an industry standard tool for distributed hyperparameter tuning. Ray Tune includes the latest hyperparameter search algorithms, integrates with various analysis libraries, and natively supports distributed training through Ray’s distributed machine learning engine.
In this tutorial, we will show you how to integrate Ray Tune into your PyTorch training workflow. We will extend this tutorial from the PyTorch documentation for training a CIFAR10 image classifier.
As you will see, we only need to add some slight modifications. In particular, we need to
wrap data loading and training in functions,
make some network parameters configurable,
add checkpointing (optional),
and define the search space for the model tuning
To run this tutorial, please make sure the following packages are installed:
ray[tune]: Distributed hyperparameter tuning librarytorchvision: For the data transformers
Setup / Imports#
Let’s start with the imports:
from functools import partial
import os
import tempfile
from pathlib import Path
import torch
import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim
from torch.utils.data import random_split
import torchvision
import torchvision.transforms as transforms
from ray import tune
from ray import train
from ray.train import Checkpoint, get_checkpoint
from ray.tune.schedulers import ASHAScheduler
import ray.cloudpickle as pickle
Most of the imports are needed for building the PyTorch model. Only the last imports are for Ray Tune.
Data loaders#
We wrap the data loaders in their own function and pass a global data directory. This way we can share a data directory between different trials.
def load_data(data_dir="./data"):
transform = transforms.Compose(
[transforms.ToTensor(), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))]
)
trainset = torchvision.datasets.CIFAR10(
root=data_dir, train=True, download=True, transform=transform
)
testset = torchvision.datasets.CIFAR10(
root=data_dir, train=False, download=True, transform=transform
)
return trainset, testset
Configurable neural network#
We can only tune those parameters that are configurable. In this example, we can specify the layer sizes of the fully connected layers:
class Net(nn.Module):
def __init__(self, l1=120, l2=84):
super(Net, self).__init__()
self.conv1 = nn.Conv2d(3, 6, 5)
self.pool = nn.MaxPool2d(2, 2)
self.conv2 = nn.Conv2d(6, 16, 5)
self.fc1 = nn.Linear(16 * 5 * 5, l1)
self.fc2 = nn.Linear(l1, l2)
self.fc3 = nn.Linear(l2, 10)
def forward(self, x):
x = self.pool(F.relu(self.conv1(x)))
x = self.pool(F.relu(self.conv2(x)))
x = torch.flatten(x, 1) # flatten all dimensions except batch
x = F.relu(self.fc1(x))
x = F.relu(self.fc2(x))
x = self.fc3(x)
return x
The train function#
Now it gets interesting, because we introduce some changes to the example from the PyTorch documentation.
We wrap the training script in a function train_cifar(config, data_dir=None).
The config parameter will receive the hyperparameters we would like to
train with. The data_dir specifies the directory where we load and store the data,
so that multiple runs can share the same data source.
We also load the model and optimizer state at the start of the run, if a checkpoint
is provided. Further down in this tutorial you will find information on how
to save the checkpoint and what it is used for.
net = Net(config["l1"], config["l2"])
checkpoint = get_checkpoint()
if checkpoint:
with checkpoint.as_directory() as checkpoint_dir:
data_path = Path(checkpoint_dir) / "data.pkl"
with open(data_path, "rb") as fp:
checkpoint_state = pickle.load(fp)
start_epoch = checkpoint_state["epoch"]
net.load_state_dict(checkpoint_state["net_state_dict"])
optimizer.load_state_dict(checkpoint_state["optimizer_state_dict"])
else:
start_epoch = 0
The learning rate of the optimizer is made configurable, too:
optimizer = optim.SGD(net.parameters(), lr=config["lr"], momentum=0.9)
We also split the training data into a training and validation subset. We thus train on 80% of the data and calculate the validation loss on the remaining 20%. The batch sizes with which we iterate through the training and test sets are configurable as well.
Adding (multi) GPU support with DataParallel#
Image classification benefits largely from GPUs. Luckily, we can continue to use
PyTorch’s abstractions in Ray Tune. Thus, we can wrap our model in nn.DataParallel
to support data parallel training on multiple GPUs:
device = "cpu"
if torch.cuda.is_available():
device = "cuda:0"
if torch.cuda.device_count() > 1:
net = nn.DataParallel(net)
net.to(device)
By using a device variable we make sure that training also works when we have
no GPUs available. PyTorch requires us to send our data to the GPU memory explicitly,
like this:
for i, data in enumerate(trainloader, 0):
inputs, labels = data
inputs, labels = inputs.to(device), labels.to(device)
The code now supports training on CPUs, on a single GPU, and on multiple GPUs. Notably, Ray also supports fractional GPUs so we can share GPUs among trials, as long as the model still fits on the GPU memory. We’ll come back to that later.
Communicating with Ray Tune#
The most interesting part is the communication with Ray Tune:
checkpoint_data = {
"epoch": epoch,
"net_state_dict": net.state_dict(),
"optimizer_state_dict": optimizer.state_dict(),
}
with tempfile.TemporaryDirectory() as checkpoint_dir:
data_path = Path(checkpoint_dir) / "data.pkl"
with open(data_path, "wb") as fp:
pickle.dump(checkpoint_data, fp)
checkpoint = Checkpoint.from_directory(checkpoint_dir)
train.report(
{"loss": val_loss / val_steps, "accuracy": correct / total},
checkpoint=checkpoint,
)
Here we first save a checkpoint and then report some metrics back to Ray Tune. Specifically, we send the validation loss and accuracy back to Ray Tune. Ray Tune can then use these metrics to decide which hyperparameter configuration lead to the best results. These metrics can also be used to stop bad performing trials early in order to avoid wasting resources on those trials.
The checkpoint saving is optional, however, it is necessary if we wanted to use advanced schedulers like Population Based Training. Also, by saving the checkpoint we can later load the trained models and validate them on a test set. Lastly, saving checkpoints is useful for fault tolerance, and it allows us to interrupt training and continue training later.
Full training function#
The full code example looks like this:
def train_cifar(config, data_dir=None):
net = Net(config["l1"], config["l2"])
device = "cpu"
if torch.cuda.is_available():
device = "cuda:0"
if torch.cuda.device_count() > 1:
net = nn.DataParallel(net)
net.to(device)
criterion = nn.CrossEntropyLoss()
optimizer = optim.SGD(net.parameters(), lr=config["lr"], momentum=0.9)
checkpoint = get_checkpoint()
if checkpoint:
with checkpoint.as_directory() as checkpoint_dir:
data_path = Path(checkpoint_dir) / "data.pkl"
with open(data_path, "rb") as fp:
checkpoint_state = pickle.load(fp)
start_epoch = checkpoint_state["epoch"]
net.load_state_dict(checkpoint_state["net_state_dict"])
optimizer.load_state_dict(checkpoint_state["optimizer_state_dict"])
else:
start_epoch = 0
trainset, testset = load_data(data_dir)
test_abs = int(len(trainset) * 0.8)
train_subset, val_subset = random_split(
trainset, [test_abs, len(trainset) - test_abs]
)
trainloader = torch.utils.data.DataLoader(
train_subset, batch_size=int(config["batch_size"]), shuffle=True, num_workers=8
)
valloader = torch.utils.data.DataLoader(
val_subset, batch_size=int(config["batch_size"]), shuffle=True, num_workers=8
)
for epoch in range(start_epoch, 10): # loop over the dataset multiple times
running_loss = 0.0
epoch_steps = 0
for i, data in enumerate(trainloader, 0):
# get the inputs; data is a list of [inputs, labels]
inputs, labels = data
inputs, labels = inputs.to(device), labels.to(device)
# zero the parameter gradients
optimizer.zero_grad()
# forward + backward + optimize
outputs = net(inputs)
loss = criterion(outputs, labels)
loss.backward()
optimizer.step()
# print statistics
running_loss += loss.item()
epoch_steps += 1
if i % 2000 == 1999: # print every 2000 mini-batches
print(
"[%d, %5d] loss: %.3f"
% (epoch + 1, i + 1, running_loss / epoch_steps)
)
running_loss = 0.0
# Validation loss
val_loss = 0.0
val_steps = 0
total = 0
correct = 0
for i, data in enumerate(valloader, 0):
with torch.no_grad():
inputs, labels = data
inputs, labels = inputs.to(device), labels.to(device)
outputs = net(inputs)
_, predicted = torch.max(outputs.data, 1)
total += labels.size(0)
correct += (predicted == labels).sum().item()
loss = criterion(outputs, labels)
val_loss += loss.cpu().numpy()
val_steps += 1
checkpoint_data = {
"epoch": epoch,
"net_state_dict": net.state_dict(),
"optimizer_state_dict": optimizer.state_dict(),
}
with tempfile.TemporaryDirectory() as checkpoint_dir:
data_path = Path(checkpoint_dir) / "data.pkl"
with open(data_path, "wb") as fp:
pickle.dump(checkpoint_data, fp)
checkpoint = Checkpoint.from_directory(checkpoint_dir)
train.report(
{"loss": val_loss / val_steps, "accuracy": correct / total},
checkpoint=checkpoint,
)
print("Finished Training")
As you can see, most of the code is adapted directly from the original example.
Test set accuracy#
Commonly the performance of a machine learning model is tested on a hold-out test set with data that has not been used for training the model. We also wrap this in a function:
def test_accuracy(net, device="cpu"):
trainset, testset = load_data()
testloader = torch.utils.data.DataLoader(
testset, batch_size=4, shuffle=False, num_workers=2
)
correct = 0
total = 0
with torch.no_grad():
for data in testloader:
images, labels = data
images, labels = images.to(device), labels.to(device)
outputs = net(images)
_, predicted = torch.max(outputs.data, 1)
total += labels.size(0)
correct += (predicted == labels).sum().item()
return correct / total
The function also expects a device parameter, so we can do the
test set validation on a GPU.
Configuring the search space#
Lastly, we need to define Ray Tune’s search space. Here is an example:
config = {
"l1": tune.choice([2 ** i for i in range(9)]),
"l2": tune.choice([2 ** i for i in range(9)]),
"lr": tune.loguniform(1e-4, 1e-1),
"batch_size": tune.choice([2, 4, 8, 16])
}
The tune.choice() accepts a list of values that are uniformly sampled from.
In this example, the l1 and l2 parameters
should be powers of 2 between 4 and 256, so either 4, 8, 16, 32, 64, 128, or 256.
The lr (learning rate) should be uniformly sampled between 0.0001 and 0.1. Lastly,
the batch size is a choice between 2, 4, 8, and 16.
At each trial, Ray Tune will now randomly sample a combination of parameters from these
search spaces. It will then train a number of models in parallel and find the best
performing one among these. We also use the ASHAScheduler which will terminate bad
performing trials early.
We wrap the train_cifar function with functools.partial to set the constant
data_dir parameter. We can also tell Ray Tune what resources should be
available for each trial:
gpus_per_trial = 2
# ...
result = tune.run(
partial(train_cifar, data_dir=data_dir),
resources_per_trial={"cpu": 8, "gpu": gpus_per_trial},
config=config,
num_samples=num_samples,
scheduler=scheduler,
checkpoint_at_end=True)
You can specify the number of CPUs, which are then available e.g.
to increase the num_workers of the PyTorch DataLoader instances. The selected
number of GPUs are made visible to PyTorch in each trial. Trials do not have access to
GPUs that haven’t been requested for them - so you don’t have to care about two trials
using the same set of resources.
Here we can also specify fractional GPUs, so something like gpus_per_trial=0.5 is
completely valid. The trials will then share GPUs among each other.
You just have to make sure that the models still fit in the GPU memory.
After training the models, we will find the best performing one and load the trained network from the checkpoint file. We then obtain the test set accuracy and report everything by printing.
The full main function looks like this:
def main(num_samples=10, max_num_epochs=10, gpus_per_trial=2):
data_dir = os.path.abspath("./data")
load_data(data_dir)
config = {
"l1": tune.choice([2**i for i in range(9)]),
"l2": tune.choice([2**i for i in range(9)]),
"lr": tune.loguniform(1e-4, 1e-1),
"batch_size": tune.choice([2, 4, 8, 16]),
}
scheduler = ASHAScheduler(
metric="loss",
mode="min",
max_t=max_num_epochs,
grace_period=1,
reduction_factor=2,
)
result = tune.run(
partial(train_cifar, data_dir=data_dir),
resources_per_trial={"cpu": 2, "gpu": gpus_per_trial},
config=config,
num_samples=num_samples,
scheduler=scheduler,
)
best_trial = result.get_best_trial("loss", "min", "last")
print(f"Best trial config: {best_trial.config}")
print(f"Best trial final validation loss: {best_trial.last_result['loss']}")
print(f"Best trial final validation accuracy: {best_trial.last_result['accuracy']}")
best_trained_model = Net(best_trial.config["l1"], best_trial.config["l2"])
device = "cpu"
if torch.cuda.is_available():
device = "cuda:0"
if gpus_per_trial > 1:
best_trained_model = nn.DataParallel(best_trained_model)
best_trained_model.to(device)
best_checkpoint = result.get_best_checkpoint(trial=best_trial, metric="accuracy", mode="max")
with best_checkpoint.as_directory() as checkpoint_dir:
data_path = Path(checkpoint_dir) / "data.pkl"
with open(data_path, "rb") as fp:
best_checkpoint_data = pickle.load(fp)
best_trained_model.load_state_dict(best_checkpoint_data["net_state_dict"])
test_acc = test_accuracy(best_trained_model, device)
print("Best trial test set accuracy: {}".format(test_acc))
if __name__ == "__main__":
# You can change the number of GPUs per trial here:
main(num_samples=10, max_num_epochs=10, gpus_per_trial=0)
0%| | 0.00/170M [00:00<?, ?B/s]
0%| | 164k/170M [00:00<01:46, 1.60MB/s]
0%| | 819k/170M [00:00<00:37, 4.48MB/s]
1%| | 1.67M/170M [00:00<00:27, 6.23MB/s]
2%|▏ | 2.62M/170M [00:00<00:22, 7.37MB/s]
2%|▏ | 3.74M/170M [00:00<00:19, 8.69MB/s]
3%|▎ | 5.05M/170M [00:00<00:16, 10.1MB/s]
4%|▍ | 6.59M/170M [00:00<00:13, 11.8MB/s]
5%|▍ | 8.32M/170M [00:00<00:12, 13.5MB/s]
6%|▌ | 10.3M/170M [00:00<00:10, 15.4MB/s]
7%|▋ | 12.4M/170M [00:01<00:09, 17.1MB/s]
9%|▊ | 14.5M/170M [00:01<00:08, 18.4MB/s]
10%|▉ | 16.5M/170M [00:01<00:08, 18.7MB/s]
11%|█ | 18.8M/170M [00:01<00:07, 19.9MB/s]
12%|█▏ | 20.9M/170M [00:01<00:07, 20.2MB/s]
13%|█▎ | 23.0M/170M [00:01<00:07, 20.4MB/s]
15%|█▍ | 25.1M/170M [00:01<00:07, 20.6MB/s]
16%|█▌ | 27.3M/170M [00:01<00:06, 20.8MB/s]
17%|█▋ | 29.5M/170M [00:01<00:06, 21.1MB/s]
19%|█▊ | 31.7M/170M [00:01<00:06, 21.4MB/s]
20%|██ | 34.4M/170M [00:02<00:05, 23.0MB/s]
22%|██▏ | 37.0M/170M [00:02<00:05, 23.8MB/s]
23%|██▎ | 39.7M/170M [00:02<00:05, 24.9MB/s]
25%|██▍ | 42.3M/170M [00:02<00:05, 25.2MB/s]
26%|██▋ | 45.1M/170M [00:02<00:04, 25.9MB/s]
28%|██▊ | 47.7M/170M [00:02<00:05, 24.1MB/s]
29%|██▉ | 50.1M/170M [00:02<00:05, 23.4MB/s]
31%|███ | 52.5M/170M [00:02<00:05, 22.7MB/s]
32%|███▏ | 54.8M/170M [00:02<00:05, 22.6MB/s]
33%|███▎ | 57.1M/170M [00:02<00:05, 22.0MB/s]
35%|███▍ | 59.6M/170M [00:03<00:04, 22.9MB/s]
37%|███▋ | 62.3M/170M [00:03<00:04, 23.8MB/s]
38%|███▊ | 64.8M/170M [00:03<00:04, 24.4MB/s]
40%|███▉ | 67.4M/170M [00:03<00:04, 24.7MB/s]
41%|████ | 70.1M/170M [00:03<00:03, 25.3MB/s]
43%|████▎ | 72.7M/170M [00:03<00:03, 25.5MB/s]
44%|████▍ | 75.6M/170M [00:03<00:03, 26.5MB/s]
46%|████▌ | 78.3M/170M [00:03<00:03, 26.5MB/s]
47%|████▋ | 81.0M/170M [00:03<00:03, 26.1MB/s]
49%|████▉ | 83.8M/170M [00:04<00:03, 26.6MB/s]
51%|█████ | 86.5M/170M [00:04<00:03, 26.3MB/s]
52%|█████▏ | 89.3M/170M [00:04<00:03, 26.9MB/s]
54%|█████▍ | 92.0M/170M [00:04<00:02, 26.6MB/s]
56%|█████▌ | 94.8M/170M [00:04<00:02, 26.7MB/s]
57%|█████▋ | 97.6M/170M [00:04<00:02, 27.1MB/s]
59%|█████▉ | 100M/170M [00:04<00:02, 26.4MB/s]
60%|██████ | 103M/170M [00:04<00:02, 26.6MB/s]
62%|██████▏ | 106M/170M [00:04<00:02, 25.3MB/s]
63%|██████▎ | 108M/170M [00:04<00:02, 23.8MB/s]
65%|██████▍ | 111M/170M [00:05<00:02, 22.9MB/s]
66%|██████▋ | 113M/170M [00:05<00:02, 22.5MB/s]
68%|██████▊ | 115M/170M [00:05<00:02, 22.7MB/s]
69%|██████▉ | 118M/170M [00:05<00:02, 24.1MB/s]
71%|███████ | 121M/170M [00:05<00:02, 24.6MB/s]
72%|███████▏ | 124M/170M [00:05<00:01, 25.3MB/s]
74%|███████▍ | 126M/170M [00:05<00:01, 25.7MB/s]
76%|███████▌ | 129M/170M [00:05<00:01, 25.9MB/s]
77%|███████▋ | 132M/170M [00:05<00:01, 26.1MB/s]
79%|███████▊ | 134M/170M [00:06<00:01, 26.2MB/s]
80%|████████ | 137M/170M [00:06<00:01, 26.6MB/s]
82%|████████▏ | 140M/170M [00:06<00:01, 26.4MB/s]
84%|████████▎ | 143M/170M [00:06<00:01, 27.0MB/s]
85%|████████▌ | 145M/170M [00:06<00:00, 26.6MB/s]
87%|████████▋ | 149M/170M [00:06<00:00, 29.1MB/s]
89%|████████▉ | 153M/170M [00:06<00:00, 31.5MB/s]
92%|█████████▏| 156M/170M [00:06<00:00, 32.3MB/s]
94%|█████████▎| 160M/170M [00:06<00:00, 33.6MB/s]
96%|█████████▌| 163M/170M [00:06<00:00, 33.8MB/s]
98%|█████████▊| 167M/170M [00:07<00:00, 34.8MB/s]
100%|██████████| 170M/170M [00:07<00:00, 24.0MB/s]
2025-08-29 17:39:37,296 WARNING services.py:1889 -- WARNING: The object store is using /tmp instead of /dev/shm because /dev/shm has only 2147467264 bytes available. This will harm performance! You may be able to free up space by deleting files in /dev/shm. If you are inside a Docker container, you can increase /dev/shm size by passing '--shm-size=10.24gb' to 'docker run' (or add it to the run_options list in a Ray cluster config). Make sure to set this to more than 30% of available RAM.
2025-08-29 17:39:37,356 INFO worker.py:1642 -- Started a local Ray instance.
2025-08-29 17:39:38,222 INFO tune.py:228 -- Initializing Ray automatically. For cluster usage or custom Ray initialization, call `ray.init(...)` before `tune.run(...)`.
2025-08-29 17:39:38,223 INFO tune.py:654 -- [output] This will use the new output engine with verbosity 2. To disable the new output and use the legacy output engine, set the environment variable RAY_AIR_NEW_OUTPUT=0. For more information, please see https://github.com/ray-project/ray/issues/36949
╭────────────────────────────────────────────────────────────────────╮
│ Configuration for experiment train_cifar_2025-08-29_17-39-38 │
├────────────────────────────────────────────────────────────────────┤
│ Search algorithm BasicVariantGenerator │
│ Scheduler AsyncHyperBandScheduler │
│ Number of trials 10 │
╰────────────────────────────────────────────────────────────────────╯
View detailed results here: /var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38
To visualize your results with TensorBoard, run: `tensorboard --logdir /var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38`
Trial status: 10 PENDING
Current time: 2025-08-29 17:39:38. Total running time: 0s
Logical resource usage: 14.0/16 CPUs, 0/1 GPUs (0.0/1.0 accelerator_type:A10G)
╭───────────────────────────────────────────────────────────────────────────────╮
│ Trial name status l1 l2 lr batch_size │
├───────────────────────────────────────────────────────────────────────────────┤
│ train_cifar_22b64_00000 PENDING 256 256 0.0111005 2 │
│ train_cifar_22b64_00001 PENDING 1 8 0.000393336 8 │
│ train_cifar_22b64_00002 PENDING 16 128 0.0173033 2 │
│ train_cifar_22b64_00003 PENDING 4 128 0.000935878 4 │
│ train_cifar_22b64_00004 PENDING 16 32 0.024711 2 │
│ train_cifar_22b64_00005 PENDING 8 128 0.0568899 8 │
│ train_cifar_22b64_00006 PENDING 2 8 0.00704518 16 │
│ train_cifar_22b64_00007 PENDING 32 8 0.00897247 8 │
│ train_cifar_22b64_00008 PENDING 8 1 0.0753007 16 │
│ train_cifar_22b64_00009 PENDING 1 16 0.00263056 4 │
╰───────────────────────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00000 started with configuration:
╭─────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00000 config │
├─────────────────────────────────────────────────┤
│ batch_size 2 │
│ l1 256 │
│ l2 256 │
│ lr 0.0111 │
╰─────────────────────────────────────────────────╯
Trial train_cifar_22b64_00007 started with configuration:
╭──────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00007 config │
├──────────────────────────────────────────────────┤
│ batch_size 8 │
│ l1 32 │
│ l2 8 │
│ lr 0.00897 │
╰──────────────────────────────────────────────────╯
Trial train_cifar_22b64_00006 started with configuration:
╭──────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00006 config │
├──────────────────────────────────────────────────┤
│ batch_size 16 │
│ l1 2 │
│ l2 8 │
│ lr 0.00705 │
╰──────────────────────────────────────────────────╯
Trial train_cifar_22b64_00004 started with configuration:
╭──────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00004 config │
├──────────────────────────────────────────────────┤
│ batch_size 2 │
│ l1 16 │
│ l2 32 │
│ lr 0.02471 │
╰──────────────────────────────────────────────────╯
Trial train_cifar_22b64_00003 started with configuration:
╭──────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00003 config │
├──────────────────────────────────────────────────┤
│ batch_size 4 │
│ l1 4 │
│ l2 128 │
│ lr 0.00094 │
╰──────────────────────────────────────────────────╯
Trial train_cifar_22b64_00001 started with configuration:
╭──────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00001 config │
├──────────────────────────────────────────────────┤
│ batch_size 8 │
│ l1 1 │
│ l2 8 │
│ lr 0.00039 │
╰──────────────────────────────────────────────────╯
Trial train_cifar_22b64_00005 started with configuration:
╭──────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00005 config │
├──────────────────────────────────────────────────┤
│ batch_size 8 │
│ l1 8 │
│ l2 128 │
│ lr 0.05689 │
╰──────────────────────────────────────────────────╯
Trial train_cifar_22b64_00002 started with configuration:
╭─────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00002 config │
├─────────────────────────────────────────────────┤
│ batch_size 2 │
│ l1 16 │
│ l2 128 │
│ lr 0.0173 │
╰─────────────────────────────────────────────────╯
(func pid=3937) [1, 2000] loss: 2.316
(func pid=3937) [1, 4000] loss: 1.165 [repeated 8x across cluster] (Ray deduplicates logs by default. Set RAY_DEDUP_LOGS=0 to disable log deduplication, or see https://docs.ray.io/en/master/ray-observability/ray-logging.html#log-deduplication for more options.)
Trial train_cifar_22b64_00006 finished iteration 1 at 2025-08-29 17:40:07. Total running time: 29s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00006 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000000 │
│ time_this_iter_s 25.19236 │
│ time_total_s 25.19236 │
│ training_iteration 1 │
│ accuracy 0.2832 │
│ loss 1.77303 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00006 saved a checkpoint for iteration 1 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00006_6_batch_size=16,l1=2,l2=8,lr=0.0070_2025-08-29_17-39-38/checkpoint_000000
(func pid=3938) Checkpoint successfully created at: Checkpoint(filesystem=local, path=/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00006_6_batch_size=16,l1=2,l2=8,lr=0.0070_2025-08-29_17-39-38/checkpoint_000000)
Trial status: 8 RUNNING | 2 PENDING
Current time: 2025-08-29 17:40:08. Total running time: 30s
Logical resource usage: 16.0/16 CPUs, 0/1 GPUs (0.0/1.0 accelerator_type:A10G)
╭──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ Trial name status l1 l2 lr batch_size iter total time (s) loss accuracy │
├──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┤
│ train_cifar_22b64_00000 RUNNING 256 256 0.0111005 2 │
│ train_cifar_22b64_00001 RUNNING 1 8 0.000393336 8 │
│ train_cifar_22b64_00002 RUNNING 16 128 0.0173033 2 │
│ train_cifar_22b64_00003 RUNNING 4 128 0.000935878 4 │
│ train_cifar_22b64_00004 RUNNING 16 32 0.024711 2 │
│ train_cifar_22b64_00005 RUNNING 8 128 0.0568899 8 │
│ train_cifar_22b64_00006 RUNNING 2 8 0.00704518 16 1 25.1924 1.77303 0.2832 │
│ train_cifar_22b64_00007 RUNNING 32 8 0.00897247 8 │
│ train_cifar_22b64_00008 PENDING 8 1 0.0753007 16 │
│ train_cifar_22b64_00009 PENDING 1 16 0.00263056 4 │
╰──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
(func pid=3937) [1, 6000] loss: 0.777 [repeated 7x across cluster]
Trial train_cifar_22b64_00001 finished iteration 1 at 2025-08-29 17:40:22. Total running time: 44s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00001 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000000 │
│ time_this_iter_s 40.05164 │
│ time_total_s 40.05164 │
│ training_iteration 1 │
│ accuracy 0.0962 │
│ loss 2.30288 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00001 saved a checkpoint for iteration 1 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00001_1_batch_size=8,l1=1,l2=8,lr=0.0004_2025-08-29_17-39-38/checkpoint_000000
Trial train_cifar_22b64_00001 completed after 1 iterations at 2025-08-29 17:40:22. Total running time: 44s
Trial train_cifar_22b64_00008 started with configuration:
╭─────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00008 config │
├─────────────────────────────────────────────────┤
│ batch_size 16 │
│ l1 8 │
│ l2 1 │
│ lr 0.0753 │
╰─────────────────────────────────────────────────╯
(func pid=3930) Checkpoint successfully created at: Checkpoint(filesystem=local, path=/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00001_1_batch_size=8,l1=1,l2=8,lr=0.0004_2025-08-29_17-39-38/checkpoint_000000)
Trial train_cifar_22b64_00005 finished iteration 1 at 2025-08-29 17:40:23. Total running time: 44s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00005 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000000 │
│ time_this_iter_s 40.26548 │
│ time_total_s 40.26548 │
│ training_iteration 1 │
│ accuracy 0.102 │
│ loss 2.318 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00005 saved a checkpoint for iteration 1 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00005_5_batch_size=8,l1=8,l2=128,lr=0.0569_2025-08-29_17-39-38/checkpoint_000000
Trial train_cifar_22b64_00005 completed after 1 iterations at 2025-08-29 17:40:23. Total running time: 44s
Trial train_cifar_22b64_00009 started with configuration:
╭──────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00009 config │
├──────────────────────────────────────────────────┤
│ batch_size 4 │
│ l1 1 │
│ l2 16 │
│ lr 0.00263 │
╰──────────────────────────────────────────────────╯
(func pid=3939) Checkpoint successfully created at: Checkpoint(filesystem=local, path=/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00005_5_batch_size=8,l1=8,l2=128,lr=0.0569_2025-08-29_17-39-38/checkpoint_000000)
(func pid=3938) [2, 2000] loss: 1.738 [repeated 4x across cluster]
Trial train_cifar_22b64_00007 finished iteration 1 at 2025-08-29 17:40:23. Total running time: 45s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00007 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000000 │
│ time_this_iter_s 41.01031 │
│ time_total_s 41.01031 │
│ training_iteration 1 │
│ accuracy 0.3983 │
│ loss 1.61823 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00007 saved a checkpoint for iteration 1 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00007_7_batch_size=8,l1=32,l2=8,lr=0.0090_2025-08-29_17-39-38/checkpoint_000000
Trial train_cifar_22b64_00006 finished iteration 2 at 2025-08-29 17:40:29. Total running time: 50s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00006 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000001 │
│ time_this_iter_s 21.3603 │
│ time_total_s 46.55266 │
│ training_iteration 2 │
│ accuracy 0.3165 │
│ loss 1.66625 │
╰────────────────────────────────────────────────────────────╯
(func pid=3938) Checkpoint successfully created at: Checkpoint(filesystem=local, path=/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00006_6_batch_size=16,l1=2,l2=8,lr=0.0070_2025-08-29_17-39-38/checkpoint_000001) [repeated 2x across cluster]
Trial train_cifar_22b64_00006 saved a checkpoint for iteration 2 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00006_6_batch_size=16,l1=2,l2=8,lr=0.0070_2025-08-29_17-39-38/checkpoint_000001
(func pid=3936) [1, 8000] loss: 0.426 [repeated 3x across cluster]
(func pid=3937) [1, 10000] loss: 0.467 [repeated 2x across cluster]
Trial status: 8 RUNNING | 2 TERMINATED
Current time: 2025-08-29 17:40:38. Total running time: 1min 0s
Logical resource usage: 16.0/16 CPUs, 0/1 GPUs (0.0/1.0 accelerator_type:A10G)
╭────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ Trial name status l1 l2 lr batch_size iter total time (s) loss accuracy │
├────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┤
│ train_cifar_22b64_00000 RUNNING 256 256 0.0111005 2 │
│ train_cifar_22b64_00002 RUNNING 16 128 0.0173033 2 │
│ train_cifar_22b64_00003 RUNNING 4 128 0.000935878 4 │
│ train_cifar_22b64_00004 RUNNING 16 32 0.024711 2 │
│ train_cifar_22b64_00006 RUNNING 2 8 0.00704518 16 2 46.5527 1.66625 0.3165 │
│ train_cifar_22b64_00007 RUNNING 32 8 0.00897247 8 1 41.0103 1.61823 0.3983 │
│ train_cifar_22b64_00008 RUNNING 8 1 0.0753007 16 │
│ train_cifar_22b64_00009 RUNNING 1 16 0.00263056 4 │
│ train_cifar_22b64_00001 TERMINATED 1 8 0.000393336 8 1 40.0516 2.30288 0.0962 │
│ train_cifar_22b64_00005 TERMINATED 8 128 0.0568899 8 1 40.2655 2.318 0.102 │
╰────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
(func pid=3929) [1, 10000] loss: 0.425 [repeated 6x across cluster]
Trial train_cifar_22b64_00008 finished iteration 1 at 2025-08-29 17:40:47. Total running time: 1min 8s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00008 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000000 │
│ time_this_iter_s 24.23592 │
│ time_total_s 24.23592 │
│ training_iteration 1 │
│ accuracy 0.099 │
│ loss 2.31265 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00008 saved a checkpoint for iteration 1 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00008_8_batch_size=16,l1=8,l2=1,lr=0.0753_2025-08-29_17-39-38/checkpoint_000000
Trial train_cifar_22b64_00008 completed after 1 iterations at 2025-08-29 17:40:47. Total running time: 1min 8s
(func pid=3930) Checkpoint successfully created at: Checkpoint(filesystem=local, path=/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00008_8_batch_size=16,l1=8,l2=1,lr=0.0753_2025-08-29_17-39-38/checkpoint_000000)
(func pid=3937) [1, 12000] loss: 0.389 [repeated 2x across cluster]
Trial train_cifar_22b64_00006 finished iteration 3 at 2025-08-29 17:40:50. Total running time: 1min 12s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00006 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000002 │
│ time_this_iter_s 21.6011 │
│ time_total_s 68.15376 │
│ training_iteration 3 │
│ accuracy 0.339 │
│ loss 1.68232 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00006 saved a checkpoint for iteration 3 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00006_6_batch_size=16,l1=2,l2=8,lr=0.0070_2025-08-29_17-39-38/checkpoint_000002
(func pid=3938) Checkpoint successfully created at: Checkpoint(filesystem=local, path=/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00006_6_batch_size=16,l1=2,l2=8,lr=0.0070_2025-08-29_17-39-38/checkpoint_000002)
(func pid=3929) [1, 12000] loss: 0.356 [repeated 4x across cluster]
Trial train_cifar_22b64_00003 finished iteration 1 at 2025-08-29 17:40:53. Total running time: 1min 15s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00003 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000000 │
│ time_this_iter_s 70.50159 │
│ time_total_s 70.50159 │
│ training_iteration 1 │
│ accuracy 0.3447 │
│ loss 1.73824 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00003 saved a checkpoint for iteration 1 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00003_3_batch_size=4,l1=4,l2=128,lr=0.0009_2025-08-29_17-39-38/checkpoint_000000
Trial train_cifar_22b64_00007 finished iteration 2 at 2025-08-29 17:41:00. Total running time: 1min 22s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00007 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000001 │
│ time_this_iter_s 36.74044 │
│ time_total_s 77.75075 │
│ training_iteration 2 │
│ accuracy 0.395 │
│ loss 1.67723 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00007 saved a checkpoint for iteration 2 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00007_7_batch_size=8,l1=32,l2=8,lr=0.0090_2025-08-29_17-39-38/checkpoint_000001
Trial train_cifar_22b64_00007 completed after 2 iterations at 2025-08-29 17:41:00. Total running time: 1min 22s
(func pid=3940) Checkpoint successfully created at: Checkpoint(filesystem=local, path=/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00007_7_batch_size=8,l1=32,l2=8,lr=0.0090_2025-08-29_17-39-38/checkpoint_000001) [repeated 2x across cluster]
(func pid=3929) [1, 14000] loss: 0.305 [repeated 4x across cluster]
Trial train_cifar_22b64_00006 finished iteration 4 at 2025-08-29 17:41:08. Total running time: 1min 30s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00006 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000003 │
│ time_this_iter_s 17.50066 │
│ time_total_s 85.65442 │
│ training_iteration 4 │
│ accuracy 0.3499 │
│ loss 1.65037 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00006 saved a checkpoint for iteration 4 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00006_6_batch_size=16,l1=2,l2=8,lr=0.0070_2025-08-29_17-39-38/checkpoint_000003
(func pid=3938) Checkpoint successfully created at: Checkpoint(filesystem=local, path=/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00006_6_batch_size=16,l1=2,l2=8,lr=0.0070_2025-08-29_17-39-38/checkpoint_000003)
Trial status: 6 RUNNING | 4 TERMINATED
Current time: 2025-08-29 17:41:08. Total running time: 1min 30s
Logical resource usage: 12.0/16 CPUs, 0/1 GPUs (0.0/1.0 accelerator_type:A10G)
╭────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ Trial name status l1 l2 lr batch_size iter total time (s) loss accuracy │
├────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┤
│ train_cifar_22b64_00000 RUNNING 256 256 0.0111005 2 │
│ train_cifar_22b64_00002 RUNNING 16 128 0.0173033 2 │
│ train_cifar_22b64_00003 RUNNING 4 128 0.000935878 4 1 70.5016 1.73824 0.3447 │
│ train_cifar_22b64_00004 RUNNING 16 32 0.024711 2 │
│ train_cifar_22b64_00006 RUNNING 2 8 0.00704518 16 4 85.6544 1.65037 0.3499 │
│ train_cifar_22b64_00009 RUNNING 1 16 0.00263056 4 │
│ train_cifar_22b64_00001 TERMINATED 1 8 0.000393336 8 1 40.0516 2.30288 0.0962 │
│ train_cifar_22b64_00005 TERMINATED 8 128 0.0568899 8 1 40.2655 2.318 0.102 │
│ train_cifar_22b64_00007 TERMINATED 32 8 0.00897247 8 2 77.7508 1.67723 0.395 │
│ train_cifar_22b64_00008 TERMINATED 8 1 0.0753007 16 1 24.2359 2.31265 0.099 │
╰────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
(func pid=3929) [1, 16000] loss: 0.264 [repeated 6x across cluster]
(func pid=3929) [1, 18000] loss: 0.239 [repeated 5x across cluster]
Trial train_cifar_22b64_00009 finished iteration 1 at 2025-08-29 17:41:23. Total running time: 1min 45s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00009 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000000 │
│ time_this_iter_s 60.3755 │
│ time_total_s 60.3755 │
│ training_iteration 1 │
│ accuracy 0.1019 │
│ loss 2.30299 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00009 saved a checkpoint for iteration 1 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00009_9_batch_size=4,l1=1,l2=16,lr=0.0026_2025-08-29_17-39-38/checkpoint_000000
Trial train_cifar_22b64_00009 completed after 1 iterations at 2025-08-29 17:41:23. Total running time: 1min 45s
(func pid=3939) Checkpoint successfully created at: Checkpoint(filesystem=local, path=/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00009_9_batch_size=4,l1=1,l2=16,lr=0.0026_2025-08-29_17-39-38/checkpoint_000000)
Trial train_cifar_22b64_00006 finished iteration 5 at 2025-08-29 17:41:24. Total running time: 1min 46s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00006 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000004 │
│ time_this_iter_s 16.07229 │
│ time_total_s 101.72671 │
│ training_iteration 5 │
│ accuracy 0.365 │
│ loss 1.61364 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00006 saved a checkpoint for iteration 5 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00006_6_batch_size=16,l1=2,l2=8,lr=0.0070_2025-08-29_17-39-38/checkpoint_000004
(func pid=3929) [1, 20000] loss: 0.216 [repeated 5x across cluster]
Trial train_cifar_22b64_00004 finished iteration 1 at 2025-08-29 17:41:33. Total running time: 1min 55s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00004 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000000 │
│ time_this_iter_s 111.03422 │
│ time_total_s 111.03422 │
│ training_iteration 1 │
│ accuracy 0.1008 │
│ loss 2.31784 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00004 saved a checkpoint for iteration 1 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00004_4_batch_size=2,l1=16,l2=32,lr=0.0247_2025-08-29_17-39-38/checkpoint_000000
Trial train_cifar_22b64_00004 completed after 1 iterations at 2025-08-29 17:41:33. Total running time: 1min 55s
(func pid=3937) Checkpoint successfully created at: Checkpoint(filesystem=local, path=/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00004_4_batch_size=2,l1=16,l2=32,lr=0.0247_2025-08-29_17-39-38/checkpoint_000000) [repeated 2x across cluster]
Trial train_cifar_22b64_00002 finished iteration 1 at 2025-08-29 17:41:34. Total running time: 1min 55s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00002 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000000 │
│ time_this_iter_s 111.19163 │
│ time_total_s 111.19163 │
│ training_iteration 1 │
│ accuracy 0.1033 │
│ loss 2.31078 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00002 saved a checkpoint for iteration 1 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00002_2_batch_size=2,l1=16,l2=128,lr=0.0173_2025-08-29_17-39-38/checkpoint_000000
Trial train_cifar_22b64_00002 completed after 1 iterations at 2025-08-29 17:41:34. Total running time: 1min 55s
(func pid=3938) [6, 2000] loss: 1.632 [repeated 2x across cluster]
Trial status: 3 RUNNING | 7 TERMINATED
Current time: 2025-08-29 17:41:38. Total running time: 2min 0s
Logical resource usage: 6.0/16 CPUs, 0/1 GPUs (0.0/1.0 accelerator_type:A10G)
╭────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ Trial name status l1 l2 lr batch_size iter total time (s) loss accuracy │
├────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┤
│ train_cifar_22b64_00000 RUNNING 256 256 0.0111005 2 │
│ train_cifar_22b64_00003 RUNNING 4 128 0.000935878 4 1 70.5016 1.73824 0.3447 │
│ train_cifar_22b64_00006 RUNNING 2 8 0.00704518 16 5 101.727 1.61364 0.365 │
│ train_cifar_22b64_00001 TERMINATED 1 8 0.000393336 8 1 40.0516 2.30288 0.0962 │
│ train_cifar_22b64_00002 TERMINATED 16 128 0.0173033 2 1 111.192 2.31078 0.1033 │
│ train_cifar_22b64_00004 TERMINATED 16 32 0.024711 2 1 111.034 2.31784 0.1008 │
│ train_cifar_22b64_00005 TERMINATED 8 128 0.0568899 8 1 40.2655 2.318 0.102 │
│ train_cifar_22b64_00007 TERMINATED 32 8 0.00897247 8 2 77.7508 1.67723 0.395 │
│ train_cifar_22b64_00008 TERMINATED 8 1 0.0753007 16 1 24.2359 2.31265 0.099 │
│ train_cifar_22b64_00009 TERMINATED 1 16 0.00263056 4 1 60.3755 2.30299 0.1019 │
╰────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00006 finished iteration 6 at 2025-08-29 17:41:39. Total running time: 2min 0s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00006 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000005 │
│ time_this_iter_s 14.68396 │
│ time_total_s 116.41067 │
│ training_iteration 6 │
│ accuracy 0.3426 │
│ loss 1.6575 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00006 saved a checkpoint for iteration 6 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00006_6_batch_size=16,l1=2,l2=8,lr=0.0070_2025-08-29_17-39-38/checkpoint_000005
(func pid=3938) Checkpoint successfully created at: Checkpoint(filesystem=local, path=/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00006_6_batch_size=16,l1=2,l2=8,lr=0.0070_2025-08-29_17-39-38/checkpoint_000005) [repeated 2x across cluster]
Trial train_cifar_22b64_00000 finished iteration 1 at 2025-08-29 17:41:40. Total running time: 2min 1s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00000 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000000 │
│ time_this_iter_s 117.56669 │
│ time_total_s 117.56669 │
│ training_iteration 1 │
│ accuracy 0.192 │
│ loss 2.1002 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00000 saved a checkpoint for iteration 1 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00000_0_batch_size=2,l1=256,l2=256,lr=0.0111_2025-08-29_17-39-38/checkpoint_000000
Trial train_cifar_22b64_00003 finished iteration 2 at 2025-08-29 17:41:42. Total running time: 2min 3s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00003 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000001 │
│ time_this_iter_s 48.74726 │
│ time_total_s 119.24885 │
│ training_iteration 2 │
│ accuracy 0.4079 │
│ loss 1.56994 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00003 saved a checkpoint for iteration 2 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00003_3_batch_size=4,l1=4,l2=128,lr=0.0009_2025-08-29_17-39-38/checkpoint_000001
(func pid=3929) [2, 2000] loss: 2.159 [repeated 2x across cluster]
Trial train_cifar_22b64_00006 finished iteration 7 at 2025-08-29 17:41:52. Total running time: 2min 14s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00006 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000006 │
│ time_this_iter_s 13.29529 │
│ time_total_s 129.70596 │
│ training_iteration 7 │
│ accuracy 0.3728 │
│ loss 1.60738 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00006 saved a checkpoint for iteration 7 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00006_6_batch_size=16,l1=2,l2=8,lr=0.0070_2025-08-29_17-39-38/checkpoint_000006
(func pid=3938) Checkpoint successfully created at: Checkpoint(filesystem=local, path=/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00006_6_batch_size=16,l1=2,l2=8,lr=0.0070_2025-08-29_17-39-38/checkpoint_000006) [repeated 3x across cluster]
(func pid=3929) [2, 4000] loss: 1.057 [repeated 3x across cluster]
(func pid=3936) [3, 6000] loss: 0.496 [repeated 2x across cluster]
Trial train_cifar_22b64_00006 finished iteration 8 at 2025-08-29 17:42:05. Total running time: 2min 27s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00006 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000007 │
│ time_this_iter_s 13.30054 │
│ time_total_s 143.0065 │
│ training_iteration 8 │
│ accuracy 0.3369 │
│ loss 1.78835 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00006 saved a checkpoint for iteration 8 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00006_6_batch_size=16,l1=2,l2=8,lr=0.0070_2025-08-29_17-39-38/checkpoint_000007
(func pid=3938) Checkpoint successfully created at: Checkpoint(filesystem=local, path=/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00006_6_batch_size=16,l1=2,l2=8,lr=0.0070_2025-08-29_17-39-38/checkpoint_000007)
(func pid=3936) [3, 8000] loss: 0.370 [repeated 3x across cluster]
Trial status: 3 RUNNING | 7 TERMINATED
Current time: 2025-08-29 17:42:08. Total running time: 2min 30s
Logical resource usage: 6.0/16 CPUs, 0/1 GPUs (0.0/1.0 accelerator_type:A10G)
╭────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ Trial name status l1 l2 lr batch_size iter total time (s) loss accuracy │
├────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┤
│ train_cifar_22b64_00000 RUNNING 256 256 0.0111005 2 1 117.567 2.1002 0.192 │
│ train_cifar_22b64_00003 RUNNING 4 128 0.000935878 4 2 119.249 1.56994 0.4079 │
│ train_cifar_22b64_00006 RUNNING 2 8 0.00704518 16 8 143.007 1.78835 0.3369 │
│ train_cifar_22b64_00001 TERMINATED 1 8 0.000393336 8 1 40.0516 2.30288 0.0962 │
│ train_cifar_22b64_00002 TERMINATED 16 128 0.0173033 2 1 111.192 2.31078 0.1033 │
│ train_cifar_22b64_00004 TERMINATED 16 32 0.024711 2 1 111.034 2.31784 0.1008 │
│ train_cifar_22b64_00005 TERMINATED 8 128 0.0568899 8 1 40.2655 2.318 0.102 │
│ train_cifar_22b64_00007 TERMINATED 32 8 0.00897247 8 2 77.7508 1.67723 0.395 │
│ train_cifar_22b64_00008 TERMINATED 8 1 0.0753007 16 1 24.2359 2.31265 0.099 │
│ train_cifar_22b64_00009 TERMINATED 1 16 0.00263056 4 1 60.3755 2.30299 0.1019 │
╰────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
(func pid=3938) [9, 2000] loss: 1.594 [repeated 2x across cluster]
Trial train_cifar_22b64_00006 finished iteration 9 at 2025-08-29 17:42:18. Total running time: 2min 39s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00006 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000008 │
│ time_this_iter_s 12.26203 │
│ time_total_s 155.26854 │
│ training_iteration 9 │
│ accuracy 0.3792 │
│ loss 1.59544 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00006 saved a checkpoint for iteration 9 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00006_6_batch_size=16,l1=2,l2=8,lr=0.0070_2025-08-29_17-39-38/checkpoint_000008
(func pid=3938) Checkpoint successfully created at: Checkpoint(filesystem=local, path=/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00006_6_batch_size=16,l1=2,l2=8,lr=0.0070_2025-08-29_17-39-38/checkpoint_000008)
Trial train_cifar_22b64_00003 finished iteration 3 at 2025-08-29 17:42:19. Total running time: 2min 41s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00003 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000002 │
│ time_this_iter_s 37.58385 │
│ time_total_s 156.8327 │
│ training_iteration 3 │
│ accuracy 0.3977 │
│ loss 1.56569 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00003 saved a checkpoint for iteration 3 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00003_3_batch_size=4,l1=4,l2=128,lr=0.0009_2025-08-29_17-39-38/checkpoint_000002
(func pid=3929) [2, 12000] loss: 0.346 [repeated 3x across cluster]
(func pid=3938) [10, 2000] loss: 1.593 [repeated 2x across cluster]
Trial train_cifar_22b64_00006 finished iteration 10 at 2025-08-29 17:42:31. Total running time: 2min 52s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00006 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000009 │
│ time_this_iter_s 13.10643 │
│ time_total_s 168.37496 │
│ training_iteration 10 │
│ accuracy 0.3413 │
│ loss 1.64004 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00006 saved a checkpoint for iteration 10 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00006_6_batch_size=16,l1=2,l2=8,lr=0.0070_2025-08-29_17-39-38/checkpoint_000009
Trial train_cifar_22b64_00006 completed after 10 iterations at 2025-08-29 17:42:31. Total running time: 2min 52s
(func pid=3938) Checkpoint successfully created at: Checkpoint(filesystem=local, path=/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00006_6_batch_size=16,l1=2,l2=8,lr=0.0070_2025-08-29_17-39-38/checkpoint_000009) [repeated 2x across cluster]
(func pid=3936) [4, 4000] loss: 0.730 [repeated 2x across cluster]
Trial status: 2 RUNNING | 8 TERMINATED
Current time: 2025-08-29 17:42:38. Total running time: 3min 0s
Logical resource usage: 4.0/16 CPUs, 0/1 GPUs (0.0/1.0 accelerator_type:A10G)
╭────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ Trial name status l1 l2 lr batch_size iter total time (s) loss accuracy │
├────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┤
│ train_cifar_22b64_00000 RUNNING 256 256 0.0111005 2 1 117.567 2.1002 0.192 │
│ train_cifar_22b64_00003 RUNNING 4 128 0.000935878 4 3 156.833 1.56569 0.3977 │
│ train_cifar_22b64_00001 TERMINATED 1 8 0.000393336 8 1 40.0516 2.30288 0.0962 │
│ train_cifar_22b64_00002 TERMINATED 16 128 0.0173033 2 1 111.192 2.31078 0.1033 │
│ train_cifar_22b64_00004 TERMINATED 16 32 0.024711 2 1 111.034 2.31784 0.1008 │
│ train_cifar_22b64_00005 TERMINATED 8 128 0.0568899 8 1 40.2655 2.318 0.102 │
│ train_cifar_22b64_00006 TERMINATED 2 8 0.00704518 16 10 168.375 1.64004 0.3413 │
│ train_cifar_22b64_00007 TERMINATED 32 8 0.00897247 8 2 77.7508 1.67723 0.395 │
│ train_cifar_22b64_00008 TERMINATED 8 1 0.0753007 16 1 24.2359 2.31265 0.099 │
│ train_cifar_22b64_00009 TERMINATED 1 16 0.00263056 4 1 60.3755 2.30299 0.1019 │
╰────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
(func pid=3936) [4, 6000] loss: 0.481 [repeated 2x across cluster]
(func pid=3936) [4, 8000] loss: 0.366 [repeated 2x across cluster]
(func pid=3936) [4, 10000] loss: 0.277 [repeated 2x across cluster]
Trial train_cifar_22b64_00003 finished iteration 4 at 2025-08-29 17:42:56. Total running time: 3min 18s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00003 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000003 │
│ time_this_iter_s 37.2197 │
│ time_total_s 194.0524 │
│ training_iteration 4 │
│ accuracy 0.4629 │
│ loss 1.45095 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00003 saved a checkpoint for iteration 4 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00003_3_batch_size=4,l1=4,l2=128,lr=0.0009_2025-08-29_17-39-38/checkpoint_000003
(func pid=3936) Checkpoint successfully created at: Checkpoint(filesystem=local, path=/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00003_3_batch_size=4,l1=4,l2=128,lr=0.0009_2025-08-29_17-39-38/checkpoint_000003)
Trial train_cifar_22b64_00000 finished iteration 2 at 2025-08-29 17:42:57. Total running time: 3min 19s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00000 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000001 │
│ time_this_iter_s 77.24222 │
│ time_total_s 194.80892 │
│ training_iteration 2 │
│ accuracy 0.2174 │
│ loss 2.03806 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00000 saved a checkpoint for iteration 2 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00000_0_batch_size=2,l1=256,l2=256,lr=0.0111_2025-08-29_17-39-38/checkpoint_000001
Trial train_cifar_22b64_00000 completed after 2 iterations at 2025-08-29 17:42:57. Total running time: 3min 19s
(func pid=3929) Checkpoint successfully created at: Checkpoint(filesystem=local, path=/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00000_0_batch_size=2,l1=256,l2=256,lr=0.0111_2025-08-29_17-39-38/checkpoint_000001)
(func pid=3936) [5, 2000] loss: 1.365
(func pid=3936) [5, 4000] loss: 0.682
Trial status: 9 TERMINATED | 1 RUNNING
Current time: 2025-08-29 17:43:08. Total running time: 3min 30s
Logical resource usage: 2.0/16 CPUs, 0/1 GPUs (0.0/1.0 accelerator_type:A10G)
╭────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ Trial name status l1 l2 lr batch_size iter total time (s) loss accuracy │
├────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┤
│ train_cifar_22b64_00003 RUNNING 4 128 0.000935878 4 4 194.052 1.45095 0.4629 │
│ train_cifar_22b64_00000 TERMINATED 256 256 0.0111005 2 2 194.809 2.03806 0.2174 │
│ train_cifar_22b64_00001 TERMINATED 1 8 0.000393336 8 1 40.0516 2.30288 0.0962 │
│ train_cifar_22b64_00002 TERMINATED 16 128 0.0173033 2 1 111.192 2.31078 0.1033 │
│ train_cifar_22b64_00004 TERMINATED 16 32 0.024711 2 1 111.034 2.31784 0.1008 │
│ train_cifar_22b64_00005 TERMINATED 8 128 0.0568899 8 1 40.2655 2.318 0.102 │
│ train_cifar_22b64_00006 TERMINATED 2 8 0.00704518 16 10 168.375 1.64004 0.3413 │
│ train_cifar_22b64_00007 TERMINATED 32 8 0.00897247 8 2 77.7508 1.67723 0.395 │
│ train_cifar_22b64_00008 TERMINATED 8 1 0.0753007 16 1 24.2359 2.31265 0.099 │
│ train_cifar_22b64_00009 TERMINATED 1 16 0.00263056 4 1 60.3755 2.30299 0.1019 │
╰────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
(func pid=3936) [5, 6000] loss: 0.452
(func pid=3936) [5, 8000] loss: 0.338
(func pid=3936) [5, 10000] loss: 0.266
Trial train_cifar_22b64_00003 finished iteration 5 at 2025-08-29 17:43:28. Total running time: 3min 50s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00003 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000004 │
│ time_this_iter_s 31.85284 │
│ time_total_s 225.90524 │
│ training_iteration 5 │
│ accuracy 0.514 │
│ loss 1.33412 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00003 saved a checkpoint for iteration 5 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00003_3_batch_size=4,l1=4,l2=128,lr=0.0009_2025-08-29_17-39-38/checkpoint_000004
(func pid=3936) Checkpoint successfully created at: Checkpoint(filesystem=local, path=/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00003_3_batch_size=4,l1=4,l2=128,lr=0.0009_2025-08-29_17-39-38/checkpoint_000004)
(func pid=3936) [6, 2000] loss: 1.321
Trial status: 9 TERMINATED | 1 RUNNING
Current time: 2025-08-29 17:43:38. Total running time: 4min 0s
Logical resource usage: 2.0/16 CPUs, 0/1 GPUs (0.0/1.0 accelerator_type:A10G)
╭────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ Trial name status l1 l2 lr batch_size iter total time (s) loss accuracy │
├────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┤
│ train_cifar_22b64_00003 RUNNING 4 128 0.000935878 4 5 225.905 1.33412 0.514 │
│ train_cifar_22b64_00000 TERMINATED 256 256 0.0111005 2 2 194.809 2.03806 0.2174 │
│ train_cifar_22b64_00001 TERMINATED 1 8 0.000393336 8 1 40.0516 2.30288 0.0962 │
│ train_cifar_22b64_00002 TERMINATED 16 128 0.0173033 2 1 111.192 2.31078 0.1033 │
│ train_cifar_22b64_00004 TERMINATED 16 32 0.024711 2 1 111.034 2.31784 0.1008 │
│ train_cifar_22b64_00005 TERMINATED 8 128 0.0568899 8 1 40.2655 2.318 0.102 │
│ train_cifar_22b64_00006 TERMINATED 2 8 0.00704518 16 10 168.375 1.64004 0.3413 │
│ train_cifar_22b64_00007 TERMINATED 32 8 0.00897247 8 2 77.7508 1.67723 0.395 │
│ train_cifar_22b64_00008 TERMINATED 8 1 0.0753007 16 1 24.2359 2.31265 0.099 │
│ train_cifar_22b64_00009 TERMINATED 1 16 0.00263056 4 1 60.3755 2.30299 0.1019 │
╰────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
(func pid=3936) [6, 4000] loss: 0.657
(func pid=3936) [6, 6000] loss: 0.439
(func pid=3936) [6, 8000] loss: 0.324
(func pid=3936) [6, 10000] loss: 0.261
Trial train_cifar_22b64_00003 finished iteration 6 at 2025-08-29 17:44:00. Total running time: 4min 22s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00003 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000005 │
│ time_this_iter_s 32.05614 │
│ time_total_s 257.96138 │
│ training_iteration 6 │
│ accuracy 0.5218 │
│ loss 1.30214 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00003 saved a checkpoint for iteration 6 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00003_3_batch_size=4,l1=4,l2=128,lr=0.0009_2025-08-29_17-39-38/checkpoint_000005
(func pid=3936) Checkpoint successfully created at: Checkpoint(filesystem=local, path=/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00003_3_batch_size=4,l1=4,l2=128,lr=0.0009_2025-08-29_17-39-38/checkpoint_000005)
(func pid=3936) [7, 2000] loss: 1.277
Trial status: 9 TERMINATED | 1 RUNNING
Current time: 2025-08-29 17:44:09. Total running time: 4min 30s
Logical resource usage: 2.0/16 CPUs, 0/1 GPUs (0.0/1.0 accelerator_type:A10G)
╭────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ Trial name status l1 l2 lr batch_size iter total time (s) loss accuracy │
├────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┤
│ train_cifar_22b64_00003 RUNNING 4 128 0.000935878 4 6 257.961 1.30214 0.5218 │
│ train_cifar_22b64_00000 TERMINATED 256 256 0.0111005 2 2 194.809 2.03806 0.2174 │
│ train_cifar_22b64_00001 TERMINATED 1 8 0.000393336 8 1 40.0516 2.30288 0.0962 │
│ train_cifar_22b64_00002 TERMINATED 16 128 0.0173033 2 1 111.192 2.31078 0.1033 │
│ train_cifar_22b64_00004 TERMINATED 16 32 0.024711 2 1 111.034 2.31784 0.1008 │
│ train_cifar_22b64_00005 TERMINATED 8 128 0.0568899 8 1 40.2655 2.318 0.102 │
│ train_cifar_22b64_00006 TERMINATED 2 8 0.00704518 16 10 168.375 1.64004 0.3413 │
│ train_cifar_22b64_00007 TERMINATED 32 8 0.00897247 8 2 77.7508 1.67723 0.395 │
│ train_cifar_22b64_00008 TERMINATED 8 1 0.0753007 16 1 24.2359 2.31265 0.099 │
│ train_cifar_22b64_00009 TERMINATED 1 16 0.00263056 4 1 60.3755 2.30299 0.1019 │
╰────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
(func pid=3936) [7, 4000] loss: 0.652
(func pid=3936) [7, 6000] loss: 0.429
(func pid=3936) [7, 8000] loss: 0.324
(func pid=3936) [7, 10000] loss: 0.256
Trial train_cifar_22b64_00003 finished iteration 7 at 2025-08-29 17:44:32. Total running time: 4min 54s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00003 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000006 │
│ time_this_iter_s 31.67513 │
│ time_total_s 289.63651 │
│ training_iteration 7 │
│ accuracy 0.5239 │
│ loss 1.33506 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00003 saved a checkpoint for iteration 7 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00003_3_batch_size=4,l1=4,l2=128,lr=0.0009_2025-08-29_17-39-38/checkpoint_000006
(func pid=3936) Checkpoint successfully created at: Checkpoint(filesystem=local, path=/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00003_3_batch_size=4,l1=4,l2=128,lr=0.0009_2025-08-29_17-39-38/checkpoint_000006)
(func pid=3936) [8, 2000] loss: 1.265
Trial status: 9 TERMINATED | 1 RUNNING
Current time: 2025-08-29 17:44:39. Total running time: 5min 0s
Logical resource usage: 2.0/16 CPUs, 0/1 GPUs (0.0/1.0 accelerator_type:A10G)
╭────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ Trial name status l1 l2 lr batch_size iter total time (s) loss accuracy │
├────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┤
│ train_cifar_22b64_00003 RUNNING 4 128 0.000935878 4 7 289.637 1.33506 0.5239 │
│ train_cifar_22b64_00000 TERMINATED 256 256 0.0111005 2 2 194.809 2.03806 0.2174 │
│ train_cifar_22b64_00001 TERMINATED 1 8 0.000393336 8 1 40.0516 2.30288 0.0962 │
│ train_cifar_22b64_00002 TERMINATED 16 128 0.0173033 2 1 111.192 2.31078 0.1033 │
│ train_cifar_22b64_00004 TERMINATED 16 32 0.024711 2 1 111.034 2.31784 0.1008 │
│ train_cifar_22b64_00005 TERMINATED 8 128 0.0568899 8 1 40.2655 2.318 0.102 │
│ train_cifar_22b64_00006 TERMINATED 2 8 0.00704518 16 10 168.375 1.64004 0.3413 │
│ train_cifar_22b64_00007 TERMINATED 32 8 0.00897247 8 2 77.7508 1.67723 0.395 │
│ train_cifar_22b64_00008 TERMINATED 8 1 0.0753007 16 1 24.2359 2.31265 0.099 │
│ train_cifar_22b64_00009 TERMINATED 1 16 0.00263056 4 1 60.3755 2.30299 0.1019 │
╰────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
(func pid=3936) [8, 4000] loss: 0.628
(func pid=3936) [8, 6000] loss: 0.424
(func pid=3936) [8, 8000] loss: 0.319
(func pid=3936) [8, 10000] loss: 0.256
Trial train_cifar_22b64_00003 finished iteration 8 at 2025-08-29 17:45:04. Total running time: 5min 25s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00003 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000007 │
│ time_this_iter_s 31.71952 │
│ time_total_s 321.35603 │
│ training_iteration 8 │
│ accuracy 0.5309 │
│ loss 1.3138 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00003 saved a checkpoint for iteration 8 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00003_3_batch_size=4,l1=4,l2=128,lr=0.0009_2025-08-29_17-39-38/checkpoint_000007
(func pid=3936) Checkpoint successfully created at: Checkpoint(filesystem=local, path=/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00003_3_batch_size=4,l1=4,l2=128,lr=0.0009_2025-08-29_17-39-38/checkpoint_000007)
Trial status: 9 TERMINATED | 1 RUNNING
Current time: 2025-08-29 17:45:09. Total running time: 5min 30s
Logical resource usage: 2.0/16 CPUs, 0/1 GPUs (0.0/1.0 accelerator_type:A10G)
╭────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ Trial name status l1 l2 lr batch_size iter total time (s) loss accuracy │
├────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┤
│ train_cifar_22b64_00003 RUNNING 4 128 0.000935878 4 8 321.356 1.3138 0.5309 │
│ train_cifar_22b64_00000 TERMINATED 256 256 0.0111005 2 2 194.809 2.03806 0.2174 │
│ train_cifar_22b64_00001 TERMINATED 1 8 0.000393336 8 1 40.0516 2.30288 0.0962 │
│ train_cifar_22b64_00002 TERMINATED 16 128 0.0173033 2 1 111.192 2.31078 0.1033 │
│ train_cifar_22b64_00004 TERMINATED 16 32 0.024711 2 1 111.034 2.31784 0.1008 │
│ train_cifar_22b64_00005 TERMINATED 8 128 0.0568899 8 1 40.2655 2.318 0.102 │
│ train_cifar_22b64_00006 TERMINATED 2 8 0.00704518 16 10 168.375 1.64004 0.3413 │
│ train_cifar_22b64_00007 TERMINATED 32 8 0.00897247 8 2 77.7508 1.67723 0.395 │
│ train_cifar_22b64_00008 TERMINATED 8 1 0.0753007 16 1 24.2359 2.31265 0.099 │
│ train_cifar_22b64_00009 TERMINATED 1 16 0.00263056 4 1 60.3755 2.30299 0.1019 │
╰────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
(func pid=3936) [9, 2000] loss: 1.256
(func pid=3936) [9, 4000] loss: 0.622
(func pid=3936) [9, 6000] loss: 0.420
(func pid=3936) [9, 8000] loss: 0.319
(func pid=3936) [9, 10000] loss: 0.252
Trial train_cifar_22b64_00003 finished iteration 9 at 2025-08-29 17:45:36. Total running time: 5min 57s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00003 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000008 │
│ time_this_iter_s 31.97831 │
│ time_total_s 353.33434 │
│ training_iteration 9 │
│ accuracy 0.5354 │
│ loss 1.29085 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00003 saved a checkpoint for iteration 9 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00003_3_batch_size=4,l1=4,l2=128,lr=0.0009_2025-08-29_17-39-38/checkpoint_000008
(func pid=3936) Checkpoint successfully created at: Checkpoint(filesystem=local, path=/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00003_3_batch_size=4,l1=4,l2=128,lr=0.0009_2025-08-29_17-39-38/checkpoint_000008)
Trial status: 9 TERMINATED | 1 RUNNING
Current time: 2025-08-29 17:45:39. Total running time: 6min 0s
Logical resource usage: 2.0/16 CPUs, 0/1 GPUs (0.0/1.0 accelerator_type:A10G)
╭────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ Trial name status l1 l2 lr batch_size iter total time (s) loss accuracy │
├────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┤
│ train_cifar_22b64_00003 RUNNING 4 128 0.000935878 4 9 353.334 1.29085 0.5354 │
│ train_cifar_22b64_00000 TERMINATED 256 256 0.0111005 2 2 194.809 2.03806 0.2174 │
│ train_cifar_22b64_00001 TERMINATED 1 8 0.000393336 8 1 40.0516 2.30288 0.0962 │
│ train_cifar_22b64_00002 TERMINATED 16 128 0.0173033 2 1 111.192 2.31078 0.1033 │
│ train_cifar_22b64_00004 TERMINATED 16 32 0.024711 2 1 111.034 2.31784 0.1008 │
│ train_cifar_22b64_00005 TERMINATED 8 128 0.0568899 8 1 40.2655 2.318 0.102 │
│ train_cifar_22b64_00006 TERMINATED 2 8 0.00704518 16 10 168.375 1.64004 0.3413 │
│ train_cifar_22b64_00007 TERMINATED 32 8 0.00897247 8 2 77.7508 1.67723 0.395 │
│ train_cifar_22b64_00008 TERMINATED 8 1 0.0753007 16 1 24.2359 2.31265 0.099 │
│ train_cifar_22b64_00009 TERMINATED 1 16 0.00263056 4 1 60.3755 2.30299 0.1019 │
╰────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
(func pid=3936) [10, 2000] loss: 1.225
(func pid=3936) [10, 4000] loss: 0.634
(func pid=3936) [10, 6000] loss: 0.414
(func pid=3936) [10, 8000] loss: 0.312
(func pid=3936) [10, 10000] loss: 0.248
Trial train_cifar_22b64_00003 finished iteration 10 at 2025-08-29 17:46:07. Total running time: 6min 29s
╭────────────────────────────────────────────────────────────╮
│ Trial train_cifar_22b64_00003 result │
├────────────────────────────────────────────────────────────┤
│ checkpoint_dir_name checkpoint_000009 │
│ time_this_iter_s 31.58812 │
│ time_total_s 384.92246 │
│ training_iteration 10 │
│ accuracy 0.5358 │
│ loss 1.29438 │
╰────────────────────────────────────────────────────────────╯
Trial train_cifar_22b64_00003 saved a checkpoint for iteration 10 at: (local)/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00003_3_batch_size=4,l1=4,l2=128,lr=0.0009_2025-08-29_17-39-38/checkpoint_000009
Trial train_cifar_22b64_00003 completed after 10 iterations at 2025-08-29 17:46:07. Total running time: 6min 29s
Trial status: 10 TERMINATED
Current time: 2025-08-29 17:46:07. Total running time: 6min 29s
Logical resource usage: 2.0/16 CPUs, 0/1 GPUs (0.0/1.0 accelerator_type:A10G)
╭────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ Trial name status l1 l2 lr batch_size iter total time (s) loss accuracy │
├────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┤
│ train_cifar_22b64_00000 TERMINATED 256 256 0.0111005 2 2 194.809 2.03806 0.2174 │
│ train_cifar_22b64_00001 TERMINATED 1 8 0.000393336 8 1 40.0516 2.30288 0.0962 │
│ train_cifar_22b64_00002 TERMINATED 16 128 0.0173033 2 1 111.192 2.31078 0.1033 │
│ train_cifar_22b64_00003 TERMINATED 4 128 0.000935878 4 10 384.922 1.29438 0.5358 │
│ train_cifar_22b64_00004 TERMINATED 16 32 0.024711 2 1 111.034 2.31784 0.1008 │
│ train_cifar_22b64_00005 TERMINATED 8 128 0.0568899 8 1 40.2655 2.318 0.102 │
│ train_cifar_22b64_00006 TERMINATED 2 8 0.00704518 16 10 168.375 1.64004 0.3413 │
│ train_cifar_22b64_00007 TERMINATED 32 8 0.00897247 8 2 77.7508 1.67723 0.395 │
│ train_cifar_22b64_00008 TERMINATED 8 1 0.0753007 16 1 24.2359 2.31265 0.099 │
│ train_cifar_22b64_00009 TERMINATED 1 16 0.00263056 4 1 60.3755 2.30299 0.1019 │
╰────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
Best trial config: {'l1': 4, 'l2': 128, 'lr': 0.0009358781867362612, 'batch_size': 4}
Best trial final validation loss: 1.2943766923382878
Best trial final validation accuracy: 0.5358
(func pid=3936) Checkpoint successfully created at: Checkpoint(filesystem=local, path=/var/lib/ci-user/ray_results/train_cifar_2025-08-29_17-39-38/train_cifar_22b64_00003_3_batch_size=4,l1=4,l2=128,lr=0.0009_2025-08-29_17-39-38/checkpoint_000009)
Best trial test set accuracy: 0.533
If you run the code, an example output could look like this:
Number of trials: 10/10 (10 TERMINATED)
+-----+--------------+------+------+-------------+--------+---------+------------+
| ... | batch_size | l1 | l2 | lr | iter | loss | accuracy |
|-----+--------------+------+------+-------------+--------+---------+------------|
| ... | 2 | 1 | 256 | 0.000668163 | 1 | 2.31479 | 0.0977 |
| ... | 4 | 64 | 8 | 0.0331514 | 1 | 2.31605 | 0.0983 |
| ... | 4 | 2 | 1 | 0.000150295 | 1 | 2.30755 | 0.1023 |
| ... | 16 | 32 | 32 | 0.0128248 | 10 | 1.66912 | 0.4391 |
| ... | 4 | 8 | 128 | 0.00464561 | 2 | 1.7316 | 0.3463 |
| ... | 8 | 256 | 8 | 0.00031556 | 1 | 2.19409 | 0.1736 |
| ... | 4 | 16 | 256 | 0.00574329 | 2 | 1.85679 | 0.3368 |
| ... | 8 | 2 | 2 | 0.00325652 | 1 | 2.30272 | 0.0984 |
| ... | 2 | 2 | 2 | 0.000342987 | 2 | 1.76044 | 0.292 |
| ... | 4 | 64 | 32 | 0.003734 | 8 | 1.53101 | 0.4761 |
+-----+--------------+------+------+-------------+--------+---------+------------+
Best trial config: {'l1': 64, 'l2': 32, 'lr': 0.0037339984519545164, 'batch_size': 4}
Best trial final validation loss: 1.5310075663924216
Best trial final validation accuracy: 0.4761
Best trial test set accuracy: 0.4737
Most trials have been stopped early in order to avoid wasting resources. The best performing trial achieved a validation accuracy of about 47%, which could be confirmed on the test set.
So that’s it! You can now tune the parameters of your PyTorch models.
Total running time of the script: (6 minutes 48.635 seconds)
