Spaces:
Sleeping
Sleeping
Update train.py
Browse files
train.py
CHANGED
|
@@ -7,7 +7,6 @@ import cv2
|
|
| 7 |
import argparse
|
| 8 |
import typing
|
| 9 |
import h5py
|
| 10 |
-
import spaces
|
| 11 |
|
| 12 |
# 解析命令行参数
|
| 13 |
def parse_opt(known=False):
|
|
@@ -177,7 +176,6 @@ def train_one_step(model, noise_image, optimizer, target_content_features, targe
|
|
| 177 |
optimizer.apply_gradients([(grad, noise_image)])
|
| 178 |
return loss
|
| 179 |
|
| 180 |
-
@spaces.GPU
|
| 181 |
def main(content_img, style_img, epochs, step_per_epoch, learning_rate, content_loss_factor, style_loss_factor, img_size, img_width, img_height):
|
| 182 |
global CONTENT_LOSS_FACTOR, STYLE_LOSS_FACTOR, CONTENT_IMAGE_PATH, STYLE_IMAGE_PATH, OUTPUT_DIR, EPOCHS, LEARNING_RATE, STEPS_PER_EPOCH, M, N, image_mean, image_std, IMG_WIDTH, IMG_HEIGHT
|
| 183 |
|
|
|
|
| 7 |
import argparse
|
| 8 |
import typing
|
| 9 |
import h5py
|
|
|
|
| 10 |
|
| 11 |
# 解析命令行参数
|
| 12 |
def parse_opt(known=False):
|
|
|
|
| 176 |
optimizer.apply_gradients([(grad, noise_image)])
|
| 177 |
return loss
|
| 178 |
|
|
|
|
| 179 |
def main(content_img, style_img, epochs, step_per_epoch, learning_rate, content_loss_factor, style_loss_factor, img_size, img_width, img_height):
|
| 180 |
global CONTENT_LOSS_FACTOR, STYLE_LOSS_FACTOR, CONTENT_IMAGE_PATH, STYLE_IMAGE_PATH, OUTPUT_DIR, EPOCHS, LEARNING_RATE, STEPS_PER_EPOCH, M, N, image_mean, image_std, IMG_WIDTH, IMG_HEIGHT
|
| 181 |
|