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 |
|