Spaces:
Running
Running
modify loss
Browse files- inference.py +0 -3
inference.py
CHANGED
|
@@ -120,9 +120,6 @@ class MasteringStyleTransfer:
|
|
| 120 |
total_loss.backward()
|
| 121 |
optimizer.step()
|
| 122 |
|
| 123 |
-
gc.collect()
|
| 124 |
-
torch.cuda.empty_cache()
|
| 125 |
-
|
| 126 |
return all_results, min_loss_step
|
| 127 |
|
| 128 |
def preprocess_audio(self, audio, target_sample_rate=44100, normalize=False):
|
|
|
|
| 120 |
total_loss.backward()
|
| 121 |
optimizer.step()
|
| 122 |
|
|
|
|
|
|
|
|
|
|
| 123 |
return all_results, min_loss_step
|
| 124 |
|
| 125 |
def preprocess_audio(self, audio, target_sample_rate=44100, normalize=False):
|