From f2f041181dd6c3ef73328c114cc2c2b45622f28d Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 9 Jun 2024 03:31:38 +0200 Subject: [PATCH 1/2] Refactor code for speed and clarity --- gcp/wave_pytorch_gcp.py | 1 - train.py | 1 - train_tf.py | 1 - 3 files changed, 3 deletions(-) diff --git a/gcp/wave_pytorch_gcp.py b/gcp/wave_pytorch_gcp.py index 6427e67..753b68e 100644 --- a/gcp/wave_pytorch_gcp.py +++ b/gcp/wave_pytorch_gcp.py @@ -4,7 +4,6 @@ import scipy.io import torch - from utils import * # set printoptions diff --git a/train.py b/train.py index c0c5730..a13600b 100644 --- a/train.py +++ b/train.py @@ -3,7 +3,6 @@ import scipy.io import torch.nn as nn - from utils.torch_utils import * from utils.utils import * diff --git a/train_tf.py b/train_tf.py index 9b80c23..9f3733c 100644 --- a/train_tf.py +++ b/train_tf.py @@ -6,7 +6,6 @@ import tensorflow as tf import tensorflow.contrib.eager as tfe from plotly.offline import plot - from utils.utils import * tf.enable_eager_execution() From 217c257cb00ae9df229eb87b84769069fc98d04b Mon Sep 17 00:00:00 2001 From: UltralyticsAssistant Date: Sun, 9 Jun 2024 01:32:02 +0000 Subject: [PATCH 2/2] Auto-format by https://ultralytics.com/actions --- gcp/wave_pytorch_gcp.py | 1 + train.py | 1 + train_tf.py | 1 + 3 files changed, 3 insertions(+) diff --git a/gcp/wave_pytorch_gcp.py b/gcp/wave_pytorch_gcp.py index 753b68e..6427e67 100644 --- a/gcp/wave_pytorch_gcp.py +++ b/gcp/wave_pytorch_gcp.py @@ -4,6 +4,7 @@ import scipy.io import torch + from utils import * # set printoptions diff --git a/train.py b/train.py index a13600b..c0c5730 100644 --- a/train.py +++ b/train.py @@ -3,6 +3,7 @@ import scipy.io import torch.nn as nn + from utils.torch_utils import * from utils.utils import * diff --git a/train_tf.py b/train_tf.py index 9f3733c..9b80c23 100644 --- a/train_tf.py +++ b/train_tf.py @@ -6,6 +6,7 @@ import tensorflow as tf import tensorflow.contrib.eager as tfe from plotly.offline import plot + from utils.utils import * tf.enable_eager_execution()