vhad/SoftGroup
1
0
Fork 0
mirror of https://github.com/botastic/SoftGroup.git synced 2025-10-16 11:45:42 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

remove torch.cuda.empty_cache()

Browse Source
This commit is contained in:
Thang Vu 2022-04-08 05:53:39 +00:00
parent 559e870615
commit b5cd952a9e
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
train.py
View File

@ -113,7 +113,6 @@ if __name__ == '__main__':
for i, batch in enumerate(train_loader, start=1):
data_time.update(time.time() - end)
torch.cuda.empty_cache()
utils.cosine_lr_after_step(optimizer, cfg.optimizer.lr, epoch - 1, cfg.step_epoch,
cfg.epochs)
@ -144,7 +143,7 @@ if __name__ == '__main__':
writer.add_scalar(k, v.val, current_iter)
if i % 10 == 0:
log_str = f'Epoch [{epoch}/{cfg.epochs}][{i}/{len(train_loader)}] '
log_str += f'lr: {lr:.5f}, eta: {remain_time}, mem: {get_max_memory()}, '\
log_str += f'lr: {lr:.2g}, eta: {remain_time}, mem: {get_max_memory()}, '\
f'data_time: {data_time.val:.2f}, iter_time: {iter_time.val:.2f}'
for k, v in meter_dict.items():
log_str += f', {k}: {v.val:.4f}'