Cleaned up the code

rystylee · Nov 15, 2019 · 00ef5da · 00ef5da
1 parent b97023e
commit 00ef5da
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/src/models/BigGAN.cpp b/src/models/BigGAN.cpp
@@ -134,6 +134,11 @@ void BigGAN::stepNoise(const float truncation)
     mNoiseStepCounter += 1.0f;
 }
 
+void BigGAN::randomizeNoise(const float truncation)
+{
+    mNoiseTensor = truncatedNoiseSample({ mBatchSize, mDimZ }, truncation);
+}
+
 at::Tensor BigGAN::truncatedNoiseSample(c10::IntArrayRef size, const float truncation)
 {
     at::Tensor noise = torch::zeros(size, at::TensorOptions(at::kFloat)).to(*mDevice.get());

diff --git a/src/models/BigGAN.hpp b/src/models/BigGAN.hpp
@@ -19,6 +19,7 @@ class BigGAN final : public BaseModel
     //void forward(ofFloatImage& img, const int classIndex, at::Tensor& noiseTensor, const float truncation=0.4f);
     void stepClass();
     void stepNoise(const float truncation=0.4f);
+    void randomizeNoise(const float truncation=0.4f);
 
     static inline void denormalize_(at::Tensor& tensor)
     {

diff --git a/src/models/VGG16.hpp b/src/models/VGG16.hpp
@@ -14,8 +14,6 @@ class VGG16 final : public BaseModel
 public:
     VGG16();
     void init(const std::string& modelPath);
-    //std::pair<int, float> forward(const ofFbo& inputFbo);
-    //std::pair<int, at::Tensor> forward(const ofFbo& inputFbo);
 
     template <typename T>
     T forward(const ofFbo& inputFbo);