vessl.progress
Parameter
Description
Examples
import vessl
for epoch in range(epochs):
...
# Update experiment progress every epoch
vessl.progress((epoch+1) / epochs)def train(model, device, train_loader, optimizer, epoch, start_epoch):
model.train()
loss = 0
for batch_idx, (data, label) in enumerate(train_loader):
...
# Update experiment progress every batch
vessl.progress(
((epoch+1)*batch_size + batch_idx) / (batch_size * epochs))
)Last updated