Skip to content
Snippets Groups Projects

Better support for cuda devices

Merged Better support for cuda devices
merge_hotfix into master
Merged
Michał Pogodarequested to merge
merge_hotfix into master
Viewing commit a4170706
Prev Next
Show latest version
2 files
+ 2
− 2
Compare changes
  • Side-by-side
  • Inline

Files

src/utils.py
+ 0
− 1
@@ -14,7 +14,6 @@ import pandas as pd
@@ -14,7 +14,6 @@ import pandas as pd
import torch
import torch
import torch.nn as nn
import torch.nn as nn
import yaml
import yaml
from torch import mode
from torch.optim import Optimizer
from torch.optim import Optimizer
from src.batch_loading import get_batches, get_ordered_dataframe_len
from src.batch_loading import get_batches, get_ordered_dataframe_len