File tree 3 files changed +3
-15
lines changed
3 files changed +3
-15
lines changed Original file line number Diff line number Diff line change @@ -46,11 +46,7 @@ def training(local_rank, config):
46
46
config ["cuda device name" ] = torch .cuda .get_device_name (local_rank )
47
47
48
48
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
54
50
55
51
task = Task .init ("CIFAR10-Training" , task_name = output_path .stem )
56
52
task .connect_configuration (config )
Original file line number Diff line number Diff line change @@ -42,11 +42,7 @@ def training(local_rank, config):
42
42
config ["cuda device name" ] = torch .cuda .get_device_name (local_rank )
43
43
44
44
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
50
46
51
47
task = Task .init ("CIFAR10-Training" , task_name = output_path .stem )
52
48
task .connect_configuration (config )
Original file line number Diff line number Diff line change @@ -44,11 +44,7 @@ def training(local_rank, config):
44
44
config ["cuda device name" ] = torch .cuda .get_device_name (local_rank )
45
45
46
46
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
52
48
53
49
task = Task .init ("IMDB-Training" , task_name = output_path .stem )
54
50
task .connect_configuration (config )
You can’t perform that action at this time.
0 commit comments