Skip to content

Commit

Permalink
Removed Trains BC from examples
Browse files Browse the repository at this point in the history
  • Loading branch information
vfdev-5 authored Aug 19, 2023
1 parent 932ddd4 commit aea1f45
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 15 deletions.
6 changes: 1 addition & 5 deletions examples/cifar10/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,11 +46,7 @@ def training(local_rank, config):
config["cuda device name"] = torch.cuda.get_device_name(local_rank)

if config["with_clearml"]:
try:
from clearml import Task
except ImportError:
# Backwards-compatibility for legacy Trains SDK
from trains import Task
from clearml import Task

task = Task.init("CIFAR10-Training", task_name=output_path.stem)
task.connect_configuration(config)
Expand Down
6 changes: 1 addition & 5 deletions examples/cifar10_qat/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,11 +42,7 @@ def training(local_rank, config):
config["cuda device name"] = torch.cuda.get_device_name(local_rank)

if config["with_clearml"]:
try:
from clearml import Task
except ImportError:
# Backwards-compatibility for legacy Trains SDK
from trains import Task
from clearml import Task

task = Task.init("CIFAR10-Training", task_name=output_path.stem)
task.connect_configuration(config)
Expand Down
6 changes: 1 addition & 5 deletions examples/transformers/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,11 +44,7 @@ def training(local_rank, config):
config["cuda device name"] = torch.cuda.get_device_name(local_rank)

if config["with_clearml"]:
try:
from clearml import Task
except ImportError:
# Backwards-compatibility for legacy Trains SDK
from trains import Task
from clearml import Task

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

0 comments on commit aea1f45

Please sign in to comment.