Skip to content

Commit 02aab6f

Browse files
authored
Removed Trains BC from examples (#3029)
1 parent 932ddd4 commit 02aab6f

File tree

3 files changed

+3
-15
lines changed

3 files changed

+3
-15
lines changed

examples/cifar10/main.py

+1-5
Original file line numberDiff line numberDiff line change
@@ -46,11 +46,7 @@ def training(local_rank, config):
4646
config["cuda device name"] = torch.cuda.get_device_name(local_rank)
4747

4848
if config["with_clearml"]:
49-
try:
50-
from clearml import Task
51-
except ImportError:
52-
# Backwards-compatibility for legacy Trains SDK
53-
from trains import Task
49+
from clearml import Task
5450

5551
task = Task.init("CIFAR10-Training", task_name=output_path.stem)
5652
task.connect_configuration(config)

examples/cifar10_qat/main.py

+1-5
Original file line numberDiff line numberDiff line change
@@ -42,11 +42,7 @@ def training(local_rank, config):
4242
config["cuda device name"] = torch.cuda.get_device_name(local_rank)
4343

4444
if config["with_clearml"]:
45-
try:
46-
from clearml import Task
47-
except ImportError:
48-
# Backwards-compatibility for legacy Trains SDK
49-
from trains import Task
45+
from clearml import Task
5046

5147
task = Task.init("CIFAR10-Training", task_name=output_path.stem)
5248
task.connect_configuration(config)

examples/transformers/main.py

+1-5
Original file line numberDiff line numberDiff line change
@@ -44,11 +44,7 @@ def training(local_rank, config):
4444
config["cuda device name"] = torch.cuda.get_device_name(local_rank)
4545

4646
if config["with_clearml"]:
47-
try:
48-
from clearml import Task
49-
except ImportError:
50-
# Backwards-compatibility for legacy Trains SDK
51-
from trains import Task
47+
from clearml import Task
5248

5349
task = Task.init("IMDB-Training", task_name=output_path.stem)
5450
task.connect_configuration(config)

0 commit comments

Comments
 (0)