Upload 2 files
Browse files
BidirectionalTranslation/options/base_options.py
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
import spaces
|
| 2 |
import argparse
|
| 3 |
import os
|
| 4 |
from util import util
|
|
@@ -66,7 +66,7 @@ class BaseOptions():
|
|
| 66 |
self.initialized = True
|
| 67 |
return parser
|
| 68 |
|
| 69 |
-
|
| 70 |
def gather_options(self):
|
| 71 |
"""Initialize our parser with basic options (only once)."""
|
| 72 |
if not self.initialized:
|
|
@@ -116,7 +116,7 @@ class BaseOptions():
|
|
| 116 |
opt_file.write(message)
|
| 117 |
opt_file.write('\n')
|
| 118 |
|
| 119 |
-
|
| 120 |
def parse(self, model_global_path):
|
| 121 |
"""Parse options, create checkpoints directory suffix, and set up gpu device."""
|
| 122 |
self.model_global_path = model_global_path
|
|
|
|
| 1 |
+
#import spaces
|
| 2 |
import argparse
|
| 3 |
import os
|
| 4 |
from util import util
|
|
|
|
| 66 |
self.initialized = True
|
| 67 |
return parser
|
| 68 |
|
| 69 |
+
#@spaces.GPU
|
| 70 |
def gather_options(self):
|
| 71 |
"""Initialize our parser with basic options (only once)."""
|
| 72 |
if not self.initialized:
|
|
|
|
| 116 |
opt_file.write(message)
|
| 117 |
opt_file.write('\n')
|
| 118 |
|
| 119 |
+
#@spaces.GPU
|
| 120 |
def parse(self, model_global_path):
|
| 121 |
"""Parse options, create checkpoints directory suffix, and set up gpu device."""
|
| 122 |
self.model_global_path = model_global_path
|
BidirectionalTranslation/options/test_options.py
CHANGED
|
@@ -1,8 +1,8 @@
|
|
| 1 |
-
import spaces
|
| 2 |
from .base_options import BaseOptions
|
| 3 |
|
| 4 |
class TestOptions(BaseOptions):
|
| 5 |
-
|
| 6 |
def initialize(self, parser):
|
| 7 |
BaseOptions.initialize(self, parser)
|
| 8 |
|
|
|
|
| 1 |
+
#import spaces
|
| 2 |
from .base_options import BaseOptions
|
| 3 |
|
| 4 |
class TestOptions(BaseOptions):
|
| 5 |
+
#@spaces.GPU
|
| 6 |
def initialize(self, parser):
|
| 7 |
BaseOptions.initialize(self, parser)
|
| 8 |
|